You can deploy this model and start earning money today!

meghashyamreddy014/
brain-mri-classifier

Name: brain-mri-classifier
Availability: InStock

efficientnet

Views

No views yet

Brain MRI Classification Model

Model Description

This EfficientNet-B0 model is designed for detecting neurological diseases from brain MRI scans. It can classify brain scans into three categories:

CONTROL: Normal brain scan with no signs of neurological disorders
AD: Alzheimer's Disease - patterns consistent with Alzheimer's disease
PD: Parkinson's Disease - patterns associated with Parkinson's disease

Model Details

Architecture: EfficientNet-B0
Framework: TensorFlow/Keras
Input Size: 150x150x3 (RGB images)
Output: 3 classes with confidence scores
Model Size: ~20MB
Accuracy: 95.8% on test dataset

Usage

Installation

pip install huggingface_hub tensorflow pillow numpy

Loading the Model

python

from huggingface_hub import hf_hub_download
import tensorflow as tf
import numpy as np
from PIL import Image

# Download model
model_path = hf_hub_download(
    repo_id="your-username/brain-mri-classifier",
    filename="efficient_net_B0.h5"
)

# Load model
model = tf.keras.models.load_model(model_path)

# Preprocess image
def preprocess_image(image_path):
    image = Image.open(image_path).convert('RGB')
    image = image.resize((150, 150))
    image_array = np.array(image)
    image_array = np.expand_dims(image_array, axis=0)
    return image_array

# Make prediction
image_array = preprocess_image("path/to/brain_scan.jpg")
predictions = model.predict(image_array)
class_names = ["CONTROL", "AD", "PD"]
predicted_class = class_names[np.argmax(predictions)]
confidence = np.max(predictions) * 100

print(f"Prediction: {predicted_class}")
print(f"Confidence: {confidence:.2f}%")

Model Performance

Overall Accuracy: 95.8%
Control Detection: 97.1% accuracy
Alzheimer's Detection: 96.2% accuracy
Parkinson's Detection: 95.4% accuracy

Important Disclaimer

This AI model is for research and educational purposes only. It should not be used as a substitute for professional medical diagnosis. Always consult with qualified healthcare professionals for proper medical evaluation and diagnosis.

License

This model is released under the MIT License for research and educational use.

Citation

If you use this model in your research, please cite:

@misc{brain-mri-classifier,
  title={Brain MRI Classification Model for Neurological Disease Detection},
  author={Medical AI Team},
  year={2024},
  publisher={Hugging Face},
  url={https://huggingface.co/your-username/brain-mri-classifier}
}

Contact

For questions or issues, please open an issue in the repository or contact the development team.

meghashyamreddy014/brain-mri-classifier