yolov5-dnn-cpp-python icon indicating copy to clipboard operation
yolov5-dnn-cpp-python copied to clipboard

Published 20 hours ago • verified publisher icon hpc203

Does it work for own pretrained weights?

Open SFE92 opened this issue 3 years ago • 1 comments

Greetings! First off this is great work and I'm happy to maybe finally find a solution to my problem.

Does your code work for own weights as well, which I've gained by training on a custom dataset with YOLOv5? I need to export the .pt to .onnx files, so I can use my own wiehgts with OpenCV DNN.

SFE92 avatar May 20 '21 12:05 SFE92

I've tested the converter script with my own .pt model but apparently it doesn't work

diazGT94 avatar Sep 17 '21 09:09 diazGT94