tch-rs
tch-rs copied to clipboard
YOLO lib
I created a simple library for yolov5 inference. I added an example with a brief description of how to prepare a model and run it in Rust. You can add it to your examples.
I would be grateful if someone could help develop the library, in the future I think to add the ability to train yolov5 and speed up the NMS algorithm.
@igor-yusupov 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?
I'd love to help @igor-yusupov . I started a yolov5 implementation from scratch (no libs) here.. Still needs some work- but I guess I'm saying I understand yolov5.