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

上传者: 42102933 | 上传时间: 2021-04-07 21:02:45 | 文件大小: 5.47MB | 文件类型: ZIP
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) 骨

文件下载

资源详情

[{"title":"( 58 个子文件 5.47MB ) YOLOv3-Torch2TRT:将YOLOv3和YOLOv3-tiny(PyTorch版本)转换为TensorRT模型-源码","children":[{"title":"YOLOv3-Torch2TRT-master","children":[{"title":"train.py <span style='color:#111;'> 6.58KB </span>","children":null,"spread":false},{"title":"utils","children":[{"title":"utils.py <span style='color:#111;'> 14.33KB </span>","children":null,"spread":false},{"title":"datasets.py <span style='color:#111;'> 4.85KB </span>","children":null,"spread":false},{"title":"parse_config.py <span style='color:#111;'> 1.23KB </span>","children":null,"spread":false},{"title":"__init__.py <span style='color:#111;'> 0B </span>","children":null,"spread":false},{"title":"__pycache__","children":[{"title":"parse_config.cpython-36.pyc <span style='color:#111;'> 1.36KB </span>","children":null,"spread":false},{"title":"augmentations.cpython-36.pyc <span style='color:#111;'> 425B </span>","children":null,"spread":false},{"title":"logger.cpython-36.pyc <span style='color:#111;'> 1.22KB </span>","children":null,"spread":false},{"title":"datasets.cpython-36.pyc <span style='color:#111;'> 4.69KB </span>","children":null,"spread":false},{"title":"utils.cpython-36.pyc <span style='color:#111;'> 10.46KB </span>","children":null,"spread":false},{"title":"__init__.cpython-36.pyc <span style='color:#111;'> 131B </span>","children":null,"spread":false}],"spread":true},{"title":"logger.py <span style='color:#111;'> 674B </span>","children":null,"spread":false},{"title":"augmentations.py <span style='color:#111;'> 207B </span>","children":null,"spread":false}],"spread":true},{"title":"output","children":[{"title":"messi.png <span style='color:#111;'> 256.89KB </span>","children":null,"spread":false},{"title":"room.png <span style='color:#111;'> 325.88KB </span>","children":null,"spread":false},{"title":"herd_of_horses.png <span style='color:#111;'> 275.21KB </span>","children":null,"spread":false},{"title":"eagle.png <span style='color:#111;'> 316.80KB </span>","children":null,"spread":false},{"title":"dog.png <span style='color:#111;'> 341.08KB </span>","children":null,"spread":false},{"title":"street.png <span style='color:#111;'> 309.18KB </span>","children":null,"spread":false},{"title":"giraffe.png <span style='color:#111;'> 401.71KB </span>","children":null,"spread":false},{"title":"person.png <span style='color:#111;'> 191.71KB </span>","children":null,"spread":false},{"title":"field.png <span style='color:#111;'> 315.53KB </span>","children":null,"spread":false}],"spread":true},{"title":"test.py <span style='color:#111;'> 3.91KB </span>","children":null,"spread":false},{"title":"detect.py <span style='color:#111;'> 9.86KB </span>","children":null,"spread":false},{"title":"check.py <span style='color:#111;'> 766B </span>","children":null,"spread":false},{"title":"requirements.txt <span style='color:#111;'> 90B </span>","children":null,"spread":false},{"title":"weights","children":[{"title":"download_weights.sh <span style='color:#111;'> 303B </span>","children":null,"spread":false}],"spread":true},{"title":"models.py <span style='color:#111;'> 23.11KB </span>","children":null,"spread":false},{"title":"config","children":[{"title":"coco.data <span style='color:#111;'> 115B </span>","children":null,"spread":false},{"title":"create_custom_model.sh <span style='color:#111;'> 8.33KB </span>","children":null,"spread":false},{"title":"custom.data <span style='color:#111;'> 99B </span>","children":null,"spread":false},{"title":"yolov3.cfg <span style='color:#111;'> 8.14KB </span>","children":null,"spread":false},{"title":"yolov3-tiny.cfg <span style='color:#111;'> 1.98KB </span>","children":null,"spread":false}],"spread":true},{"title":"assets","children":[{"title":"traffic.png <span style='color:#111;'> 311.78KB </span>","children":null,"spread":false},{"title":"messi.png <span style='color:#111;'> 257.66KB </span>","children":null,"spread":false},{"title":"dog.png <span style='color:#111;'> 342.00KB </span>","children":null,"spread":false},{"title":"giraffe.png <span style='color:#111;'> 404.74KB </span>","children":null,"spread":false}],"spread":true},{"title":"README.md <span style='color:#111;'> 1.55KB </span>","children":null,"spread":false},{"title":"__pycache__","children":[{"title":"models.cpython-36.pyc <span style='color:#111;'> 14.68KB </span>","children":null,"spread":false},{"title":"test.cpython-36.pyc <span style='color:#111;'> 3.35KB </span>","children":null,"spread":false}],"spread":true},{"title":"data","children":[{"title":"custom","children":[{"title":"labels","children":[{"title":"train.txt <span style='color:#111;'> 34B </span>","children":null,"spread":false}],"spread":false},{"title":"train.txt <span style='color:#111;'> 29B </span>","children":null,"spread":false},{"title":"images","children":[{"title":"train.jpg <span style='color:#111;'> 110.64KB </span>","children":null,"spread":false}],"spread":false},{"title":"valid.txt <span style='color:#111;'> 29B </span>","children":null,"spread":false},{"title":"classes.names <span style='color:#111;'> 6B </span>","children":null,"spread":false}],"spread":false},{"title":"temptest","children":[{"title":"field.jpg <span style='color:#111;'> 111.21KB </span>","children":null,"spread":false}],"spread":false},{"title":"samples","children":[{"title":"room.jpg <span style='color:#111;'> 83.00KB </span>","children":null,"spread":false},{"title":"street.jpg <span style='color:#111;'> 100.36KB </span>","children":null,"spread":false},{"title":"messi.jpg <span style='color:#111;'> 123.89KB </span>","children":null,"spread":false},{"title":"person.jpg <span style='color:#111;'> 76.92KB </span>","children":null,"spread":false},{"title":"field.jpg <span style='color:#111;'> 111.21KB </span>","children":null,"spread":false},{"title":"giraffe.jpg <span style='color:#111;'> 373.99KB </span>","children":null,"spread":false},{"title":"eagle.jpg <span style='color:#111;'> 138.56KB </span>","children":null,"spread":false},{"title":"dog.jpg <span style='color:#111;'> 159.92KB </span>","children":null,"spread":false},{"title":"herd_of_horses.jpg <span style='color:#111;'> 130.37KB </span>","children":null,"spread":false}],"spread":false},{"title":"coco.names <span style='color:#111;'> 625B </span>","children":null,"spread":false},{"title":"5k.txt <span style='color:#111;'> 0B </span>","children":null,"spread":false},{"title":"get_coco_dataset.sh <span style='color:#111;'> 898B </span>","children":null,"spread":false}],"spread":false}],"spread":false}],"spread":true}]

评论信息

  • m0_46546316 :
    用户下载后在一定时间内未进行评价,系统默认好评。
    2021-08-19

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明