search

igor-yusupov / rusty-yolo

Apache License 2.0
13 stars 3 forks source link

How to convert YOLOv5 Model into a compatible file for tch #2

Closed decodethedev closed 1 year ago

decodethedev commented 1 year ago

How would I convert my Yolov5 Model into a compatible file to use with it? because it's giving me errors like this: "PytorchStreamReader failed locating file constants.pkl: file not found" Which I found out is because I'm using PyTorch exported model, so how would I convert it so I can use it?

Nic-Gould commented 1 year ago

Please see my fork here and let me know if you're successful.

igor-yusupov commented 1 year ago

@decodethedev sorry for late answer. you can find the script here: https://github.com/igor-yusupov/rusty-yolo/tree/main/examples/simple_inference

UPD: https://github.com/igor-yusupov/rusty-yolo/tree/main/models