In this project I used the MRI image dataset with tumor in the link:
https://www.kaggle.com/navoneel/brain-mri-images-for-brain-tumor-detection
Then I used the https://www.makesense.ai/ to annotate these images. I considered two class of objects: label=0 for tumor and label=1 for brain.
Then I trained the YOLO-V5 for 40 epochs and batch size=10
In the following image you van see the prediction for some images:
-
Notifications
You must be signed in to change notification settings - Fork 0
Sedighe-Raeisi/Tumor_Object_Detection
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published