1 Star 0 Fork 2

stonesui / yolov5_tensorrt

forked from hox / yolov5_tensorrt 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

yolov5_tensorrt

This is the implementation that supports the version 1.0 of yolov5s, yolov5m, yolov5l and yolov5x.
The Pytorch implementation of yolov5 version 1.0 is provided in this repos.
The latest ultralytics/yolov5 version 2.0 of https://github.com/ultralytics/yolov5 is not supported
You can see source code of yolov5s on https://github.com/wang-xinyu/tensorrtx/tree/master/yolov5
You can see Pytorch implementatuon of ultralytics/yolov5(version 1.0) on https://github.com/ultralytics/yolov5/releases/tag/v1.0

Test Environment

  1. GTX1080 / Ubuntu16.04 or GTX2080Ti / Ubuntu18.04
  2. cuda >= 10.0 , cudnn >= 7.0.0 , tensorrt7.0.0, nvinfer7.0.0, opencv3.3(I use opencv3.4)

How to Run

You should check the compile environment in the CMakeLists.txt for each folder(yolov5s,yolov5m,yolov5l,yolov5x).
Each folder has a readme inside, which explains how to run the models inside.
You can download pt files of ultralytics/yolov5(s,m,l,x)(version 1.0) on BaiduCloud, passwd:9cw1
Or you can download pt files of ultralytics/yolov5(s,m,l,x)(version 1.0) on https://github.com/ultralytics/yolov5/releases/tag/v1.0
Now, this repos only support yolov5 version 1.0.

Acknowledgments

Thanks to wang-xinyu for his great works and repos.

空文件

简介

https://github.com/AIpakchoi/yolov5_tensorrt 展开 收起
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
1
https://gitee.com/suilei2002/yolov5_tensorrt.git
git@gitee.com:suilei2002/yolov5_tensorrt.git
suilei2002
yolov5_tensorrt
yolov5_tensorrt
master

搜索帮助