DanaHan/Yolov5-in-Deepstream-5.0

not able to convert .pt to .wts

dhirajpatnaik16297 opened this issue · 0 comments

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