pytorch-YOLOv4 icon indicating copy to clipboard operation
pytorch-YOLOv4 copied to clipboard

any guess why I can't convert weight of yolo to pt extension?

Open kotran88 opened this issue 2 years ago • 0 comments

I saw https://medium.com/analytics-vidhya/train-a-custom-yolov4-tiny-object-detector-using-google-colab-b58be08c9593 article and with my custom data, it infer very well !

now I'm trying to convert my weight file to pt file so that I can use it in raspberry pi.


from tool.darknet2pytorch import Darknet
print("gogo")
WEIGHTS = Darknet("C:\\Users\\pedro\\Downloads\\pytorch-YOLOv4-master\\yolov4-tiny-custom.cfg")
WEIGHTS.load_weights("C:\\Users\pedro\\Downloads\\pytorch-YOLOv4-master\\yolov4-tiny-custom_best.weights")

I execute above python file to get pt file but I got error convalution havn't activate linear convalution havn't activate linear why is it happened even if I infer quite well on colab?

Thank you !

kotran88 avatar Sep 17 '21 14:09 kotran88