Croppie cherry detection model Β© 2024 by Alliance Bioversity & CIAT, Producers Direct and M-Omulimisa is licensed under GNU-GPLv3
General description
Ultralytics' Yolo V8 medium model fined tuned for coffee cherry detection using the Croppie coffee dataset.
Note: the low visibility/unsure class was not used for model fine tuning
The predicted numerical classes correspond to the following cherry types:
{0: "dark_brown_cherry", 1: "green_cherry", 2: "red_cherry", 3: "yellow_cherry"}
Repository structure
.
βββ images
β βββ 1688033955437.jpg # image for test
β βββ annotated_1688033955437_.jpg
βββ model_v3_202402021.pt # fine tuning of Yolo v8
βββ README.md
βββ LICENSE.txt # detailed term of the software license
βββ scripts
βββ render_results.py # helper function to annotate predictions
βββ requirements.txt # pip requirements
βββ test_script.py # test script
Demonstration
Assuming you are in the scripts
folder, you can run python3 test_script.py
. This script saves the annotated image in ../images/annotated_1688033955437.jpg
.
Make sure that the Python packages found in requirements.txt
are installed. In case they are not, simply run pip3 install -r requirements.txt
.
License
Croppie cherry detection model Β© 2024 by Alliance Bioversity & CIAT, Producers Direct and M-Omulimisa is licensed under GNU-GPLv3
This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
You should have received a copy of the GNU General Public License along with this program. If not, see https://www.gnu.org/licenses/.
The detailed terms of the license are available in the LICENSE
file in the repository.
Model tree for rgautroncgiar/croppie_coffee_ug
Base model
Ultralytics/YOLOv8Evaluation results
- [email protected](box) on croppie_coffeeself-reported0.691