介绍
对象检测算法的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 ,需要对“ export.py”进行如下修改以支持GPU:
# line 28
img = torch . zeros (( opt . batch_size , 3 , * opt . img_size )). to ( device = 'cuda' )
# line 31
model = attempt_load ( opt . weigh
1