Add script for converting YOLOv8 models to ONNX for inference

conservationtechlab / camml

A package of ML components for CTL field camera systems.

MIT License

4 stars 0 forks source link

Add script for converting YOLOv8 models to ONNX for inference #14

Closed EricE147 closed 1 year ago

EricE147 commented 1 year ago

Currently, training a YOLOv8 model produces .pt model files which cannot be directly used with OpenCV's cv2.dnn functions. We need a script that can convert these existing models to .onnx format so that we can then run inference on them. The functionality of this script could eventually be structured to work within cv2dnn.py.