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 316 forks source link

请问如何用python加载engine模型进行推理? #161

Open NOBUGW opened 2 years ago

NOBUGW commented 2 years ago

请问,已经导出kFLOAT-batch1.engine文件和libdetector.so文件,请问怎么用python加载并推理,有示例代码吗?

isomatical commented 2 years ago

I guess you can try ctypes.