tch-rs icon indicating copy to clipboard operation
tch-rs copied to clipboard

YOLO lib

Open igor-yusupov opened this issue 1 year ago • 2 comments

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 avatar Apr 30 '23 18:04 igor-yusupov

@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?

decodethedev avatar May 18 '23 11:05 decodethedev

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.

Nic-Gould avatar Jun 16 '23 01:06 Nic-Gould