WongKinYiu / ScaledYOLOv4

Scaled-YOLOv4: Scaling Cross Stage Partial Network
GNU General Public License v3.0
2.02k stars 570 forks source link

Scaled-Yolov4 in Jetson #154

Open saikrishnadas opened 3 years ago

saikrishnadas commented 3 years ago

I want to run my scaled-yolov4 model in Jetson Xavier Nx. How do I convert it to TensorRT ?

Trud09 commented 3 years ago

https://github.com/WongKinYiu/ScaledYOLOv4/issues/56

Convert it to onnx, and that repo includes a conversion to TensorRT in cpp.

saikrishnadas commented 3 years ago

Scaled-yolov4 saves as PyTorch weight right? So, no direct conversion to TensorRT? @Trud09

Trud09 commented 3 years ago

I am not 100% on the format of the pt of the repo. I have been able to convert it to the .onnx using the export_onnx.py from the linked repo in that post.

The repo then takes the onnx and should convert it to TensorRT, although I have not used this.

tjuskyzhang commented 3 years ago

I have made a yolov4-csp/p5/p6/p7-tensorrt project based on this project without using onnx https://github.com/tjuskyzhang/Scaled-YOLOv4-TensorRT

noorfaleh commented 3 years ago

I have made a yolov4-csp-tensorrt project based on this project without using onnx https://github.com/tjuskyzhang/Scaled-YOLOv4-TensorRT

this run on jetson nano ?

tjuskyzhang commented 3 years ago

I have made a yolov4-csp-tensorrt project based on this project without using onnx https://github.com/tjuskyzhang/Scaled-YOLOv4-TensorRT

this run on jetson nano ?

have been tested on TX2

noorfaleh commented 3 years ago

How can I train scaled yolov4-csp on custom dataset ?

яєgαя∂ѕ ησσя

On Apr 26, 2021, at 11:29 AM, tjuskyzhang @.***> wrote:

I have made a yolov4-csp-tensorrt project based on this project without using onnx https://github.com/tjuskyzhang/Scaled-YOLOv4-TensorRT

this run on jetson nano ?

have been tested on TX2

— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

soaibsafi commented 3 years ago

@noorfaleh, I found this notebook useful. This may help-

https://colab.research.google.com/drive/1LDmg0JRiC2N7_tx8wQoBzTB0jUZhywQr?usp=sharing