Skip to content

anseryuer/YOLOv5s-ONNX-visualization-and-deploy-on-webcamera-with-openCV-for-ripe-strawberry-detection_

Repository files navigation

ripe_strawberry_detection_YOLOv5s

Detecting the intended object in ONNX model. When running the python file, a window will pop up on the computer marking the input from the camera with squares.

The current model is for strawberry detection.

about the onnx model

detecting ripe strawberry using yolov5

The model is trained based on the YOLOv5s model and used to detect the ripe strawberries.

The 3 kinds of label, rs10, rs9 and rs8, shows the ripeness of the strawberry.

the final result is shown in the result.jpg.

result

visualize.py

After running this file you will see a window for opencv pops out and if a camera on your computer is available.

The model will run in real time(10 frames per second) and detect strawberries and mark their ripeness.

A box will be shown for each object.

Result

confusion_matrix

PR_curve

About

detecting ripe strawberry using yolov5

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published