This project leverages YOLOv11n.pt (You Only Look Once) to detect and classify brain tumors in medical images. Using deep learning and computer vision techniques, the model identifies and localizes tumors to support early detection and diagnosis efforts.
The project was deployed as an interactive Streamlit web app, allowing users to upload brain scan images and get instant detection results.
- Model: YOLOv11n (a lightweight and fast object detection model)
- Dataset: Brain tumor image dataset [https://universe.roboflow.com/fast-dzyt9/tumor_detection-gbckh/dataset/1]
- Training: Fine-tuned pre-trained weights (
YOLOv11n.pt) on tumor data - Deployment: Hosted via Streamlit for an interactive, browser-based UI
- π§ Real-time object detection optimized for brain MRI images
- π Localization & classification of brain tumors
- π Preprocessing pipeline: resizing, normalization, and augmentation
- π Evaluation metrics: Accuracy, IoU (Intersection over Union), Precision, Recall
- π― Transfer learning: Pre-trained weights fine-tuned for higher accuracy
- π₯οΈ High-performance GPU (NVIDIA RTX 3080 or better recommended)
- πΎ Minimum: 16 GB RAM, 500 GB storage
- π Python 3.9+
- π§ Required Libraries:
PyTorchUltralytics(for YOLO)Streamlit(for deployment)
- β Achieved 98% accuracy after 20 epochs of training
- π§ͺ Consistent performance across validation sets
- π Deployed successfully as a Streamlit app for real-time usage
Note: This is a coursework-based project meant for academic exploration and learning. It's not intended for clinical deployment.