Detection of Brain Tumor in MRI Images using YOLO v3 Model implemented in Pytorch

Directory Structure:

|-- dataset/
|      |-- images/
|      |     |-- train/   
|      |     |-- test/
|      |
|      |-- labels/
|            |-- train/
|            |-- test/
|
|-- source_code
|      |-- yolo_configuration.json
|      |-- model.py
|      |-- dataset.py
|      |-- loss_fn.py
|      |-- utils.py
|      |-- run.py    # entry point
|
|-- weights/
|-- image_plots/ # saved images after plotting of pred bounding boxes

Commands

• To Train the model

  python3 run.py --mode train --epoch 80
  

• To visualise predicted bboxes on Test set

  python3 run.py --mode test
  

Visualise

prediction.jpg showcases image with ground truth and predicted bounded boxes with resolution 416x416.

Resources

• Dataset: Brain Tumor Dataset
• Reference used for Yolov3 implementation: aladdinpersson