stu-yzZ / yolov5infer

2 stars 0 forks source link

windows平台使用 TensorRT 部署 PyTorch 模型

yolov5infer

修改demo文件即可

demo.py中调用本地摄像头输出图像到engine中推理,输出以视频实时检测方式。

需要注意得在自己的设备上执行onnx2trt的转换,所以该项目里的trt文件不支持在别的设备上跑通