YOLOv3 Torch2TRT:将YOLOv3和YOLOv3 tiny(PyTorch版本)转换为TensorRT模型 源码

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) 骨