DanaHan / Yolov5-in-Deepstream-5.0

Describe how to use yolov5 in Deepstream 5.0
208 stars 56 forks source link

not able to convert .pt to .wts #27

Open dhirajpatnaik16297 opened 3 years ago

dhirajpatnaik16297 commented 3 years ago

Hi I have trained a custom yolov5s model for single class and could not get it to convert into .wts but on the other hand conversion of pre-trained yolov5s model is easily converted as the steps say. Also there is no log generated for either cases but the pretrained model is converted to yolov5s.wts but the custom model is not. What shall i do? Any help as soon as possible is very much appreciated. Thanks