YoLov5-TensorRT-NMS:使用Tensorrt C ++ API实现yolov5,并集成批处理的NMSPlugin。 还提供了Python包装器-源码

上传者: 42150745 | 上传时间: 2021-09-03 15:21:28 | 文件大小: 18.88MB | 文件类型: ZIP
C++
约洛夫5 来自原始代码。 我修改了yololayer并集成了批处理的NMSPlugin。 提供了yolov5s.wts可以进行快速演示。 如何生成.wts可以参考 。 如何运行,以yolov5s为例 建立并运行 mkdir build cd build cmake .. make sudo ./yolov5 -s // serialize model to plan file i.e. 'yolov5s.engine' sudo ./yolov5 -d ../samples // deserialize plan file and run inference, the images in samples will be processed. 检查生成的图像,如下所示。 _zidane.jpg和_bus.jpg 运行Python示例,请安装Python ten

文件下载

资源详情

[{"title":"( 13 个子文件 18.88MB ) YoLov5-TensorRT-NMS:使用Tensorrt C ++ API实现yolov5,并集成批处理的NMSPlugin。 还提供了Python包装器-源码","children":[{"title":"YoLov5-TensorRT-NMS-master","children":[{"title":"yolov5.cpp <span style='color:#111;'> 30.76KB </span>","children":null,"spread":false},{"title":"gen_wts.py <span style='color:#111;'> 558B </span>","children":null,"spread":false},{"title":"common.hpp <span style='color:#111;'> 15.33KB </span>","children":null,"spread":false},{"title":"samples","children":[{"title":"bus.jpg <span style='color:#111;'> 476.01KB </span>","children":null,"spread":false},{"title":"zidane.jpg <span style='color:#111;'> 164.99KB </span>","children":null,"spread":false}],"spread":true},{"title":"yololayer.h <span style='color:#111;'> 4.00KB </span>","children":null,"spread":false},{"title":"README.md <span style='color:#111;'> 1.12KB </span>","children":null,"spread":false},{"title":"utils.h <span style='color:#111;'> 3.07KB </span>","children":null,"spread":false},{"title":"yolov5_trt.py <span style='color:#111;'> 10.45KB </span>","children":null,"spread":false},{"title":"logging.h <span style='color:#111;'> 16.56KB </span>","children":null,"spread":false},{"title":"yololayer.cu <span style='color:#111;'> 11.94KB </span>","children":null,"spread":false},{"title":"yolov5s.wts <span style='color:#111;'> 64.30MB </span>","children":null,"spread":false},{"title":"CMakeLists.txt <span style='color:#111;'> 1.11KB </span>","children":null,"spread":false}],"spread":false}],"spread":true}]

评论信息

免责申明

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