rgautroncgiar/croppie_coffee_ug

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.