Detect brain diseases using YOLOv8 with high accuracy from brain MRI scans
Report Bug
·
Request Feature
Table of Contents
For local use, please download the files in here
NOTE: Since the project uses AWS Lambda and API Gateway, local use cannot use the same, therefore you won't have a frontend
Once you've obtained the files, you can follow these next steps
-
Make a new folder and put these files in
model.pt yolo_inference.py
-
Optional: Create a virtual environment and activate it
python -m venv .venv . .venv/Scripts/activate
Some systems might be different (python3 rather python; . .venv/bin/activate rather than . .venv/Scripts/activate)
-
Install ultralytics
pip install ultralytics
-
Open yolo_inference.py and change the <image_file_path> to your MRI scan image path
from ultralytics import YOLO model = YOLO('model.pt') model.predict('<image_file_path>', save=True)
Contributions are what make the open source community such an amazing place to learn, inspire, and create. Any contributions you make are greatly appreciated.
If you have a suggestion that would make this better, please fork the repo and create a pull request. You can also simply open an issue with the tag "enhancement". Don't forget to give the project a star! Thanks again!
- Fork the Project
- Create your Feature Branch (
git checkout -b feature/AmazingFeature
) - Commit your Changes (
git commit -m 'Add some AmazingFeature'
) - Push to the Branch (
git push origin feature/AmazingFeature
) - Open a Pull Request
Duong Hoang - LinkedIn
Project Link: Github