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.
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.
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.