search

ttanzhiqiang / onnx_tensorrt_project

Support Yolov5(4.0)/Yolov5(5.0)/YoloR/YoloX/Yolov4/Yolov3/CenterNet/CenterFace/RetinaFace/Classify/Unet. use darknet/libtorch/pytorch/mxnet to onnx to tensorrt
210 stars 43 forks source link

yolov5x_fp32_batch_1.engine 是什么文件？如何生成？ #1

Closed zsffuture closed 3 years ago

zsffuture commented 3 years ago

您好，感谢您的开源工作，如题我现在已经转换得到onnx了，但是另一个文件yolov5x_fp32_batch_1.engine不知道怎么生成或者得到的，新手，希望大神能指导一下，谢谢

ttanzhiqiang commented 3 years ago

通过代码yolov5_detector.cpp将onnx生成engine文件，就可以做trt推理了

abhigoku10 commented 3 years ago

@ttanzhiqiang do you have python code to generate engine file and inference the engine file ??

ttanzhiqiang commented 3 years ago

no ,this is c++ project,you can look it ,and you write python code to generate engine file and inference the engine file