search

enazoe / yolo-tensorrt

TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.
MIT License
1.18k stars 313 forks source link

TensorFlow-Yolov3 > onnx > trt #63

Closed MuhammadAsadJaved closed 3 years ago

MuhammadAsadJaved commented 3 years ago

Hi, I am trying to convert TensorFlow-Yolov3 .ckpt or .pb weights to .onnx and then to .trt engine. and finally, run in Xavier NX for the interface.

I am able to convert .pb to .onnx but having problems converting .onnx to .trt engine. Is there any method to convert .ckpt weights directly to .trt engine? I am facing the following errors during conversion.

+Is there any WeChat group?

webwxgetmsgimg

enazoe commented 3 years ago

NO,we have qq group 1151955802，this project just support the darknet weights. And the problem is that you have a custom layer， the parser can not parse the custom layer.

MuhammadAsadJaved commented 3 years ago

@enazoe OK. Thank you. How to handle the custom layers parsing problem? Any method or suggestions?

enazoe commented 3 years ago

use the tensorrt api,write a plugin layer

MuhammadAsadJaved commented 3 years ago

OK. Thank you very much.