jveitchmichaelis / deeplabel

A cross-platform desktop image annotation tool for machine learning
207 stars 40 forks source link

Can deeplabel support Yolov5 h5 file or onnx file? #37

Open fanweiya opened 2 years ago

fanweiya commented 2 years ago

can deeplabel support Yolov5 h5 model file or onnx model file? thanks

jveitchmichaelis commented 1 year ago

Currently, model loading is done via OpenCV but I'd like to support other frameworks like torchscript (eg for Detectron2).

Newer versions of OpenCV support ONNX I think so that should be doable.