search

seanavery / yolov5-tensorrt

YOLOv5 in TensorRT
138 stars 37 forks source link

Jetson TX2 deployment #16

Open berkantay opened 3 years ago

berkantay commented 3 years ago

Hello @SeanAvery thanks for the repository. I am running my own custom detection on jetson tx2 with yolov5 repository https://github.com/ultralytics/yolov5 and I got 10-12 FPS on real time video. I want to increase the performance of my detection. Is this repository only valid for pretrained models like yolov5s.pt or can it be used for custom.pt?

seanavery commented 3 years ago

Hey @berkantay thanks dude.

10 fps

Wow that's awesome, can you send me the model? How many classes?

Once you export to onnx, you should be able to use the code. also need to adjust some post processing params.

Cheers

berkantay commented 3 years ago

Hello @SeanAvery I got 1 one class for detection and I have exported onnx model. I think I should modify your demo py according to video stream.