-
将PyTorch版本的YOLOv3和YOLOv3-tiny通过Torch2TRT工具转换为TensorRT模型，即YOLOv3-Torch2TRT
资源介绍
YOLOv3-Torch2TRT
介绍
通过torch2trt Python API将YOLOv3和YOLOv3-tiny（PyTorch版本）转换为TensorRT模型。
安装
克隆仓库
git clone https://github.com/DocF/YOLOv3-Torch2TRT.git
下载预先训练的体重
$ cd weights/
$ bash download_weights.sh
要求
需要两个特殊的Python包：
张量
火炬2trt
由于YOLO中的升采样操作，根据torch2trt API的介绍，您需要使用插件安装该版本。
安装参考： :
检查torch2trt API
python3 check.py
推理加速
加速技术：
FP16
TensorRT
以下是TITAN xp的一些结果：
型号名称
输入尺寸
FP16
整个模式*（FPS）
骨