LibTorch-YOLOv5: YOLOv5的LibTorch推理实现

简介：
简介：LibTorch-YOLOv5是YOLOv5模型的LibTorch版本，提供高效且便携的对象检测解决方案，适用于多种设备上的实时推理。 介绍对象检测算法的LibTorch推理实现。支持GPU和CPU。 依赖关系: - Ubuntu 16.04 - CUDA 10.2 - OpenCV 3.4.12 - LibTorch 1.6.0 对于 TorchScript 模型导出，请参考官方文档的相关部分。 强制更新：开发人员需要修改原始以下代码： # line 29 model.model[-1].export = False 添加GPU支持：当前的导出脚本默认情况下使用CPU。为了支持 GPU，需对 export.py 进行如下更改: # line 28 img = torch.zeros((opt.batch_size, 3, * opt.img_size)).to(device=cuda) # line 31 model = attempt_load(opt.weight)，请确保在加载模型时指定设备为GPU。