代码拉取完成,页面将自动刷新
This is tensorrt c++ api implementation of yolov5.
clone the project
git clone https://github.com/ultralytics/yolov5.git
git clone https://github.com/enazoe/yolo-tensorrt.git
get .cfg and .weights file
copy the script to yolov5 root dir and generate the cfg and weigths file.
cp yolo-tensorrt/scripts/yaml2cfg.py yolov5/
python yaml2cfg.py
mv yolov5/models/yolov5.cfg yolo-tensorrt/configs/yolov5-4.0
mv yolov5/weights/yolov5.weights yolo-tensorrt/configs/yolov5-4.0
transformed models
model | .yaml | .pt | .cfg | .weights |
---|---|---|---|---|
yolov5s-v4.0 | yolov5s.yaml | yolov5s.pt | yolov5s.cfg | yolov5s.weights |
yolov5m-v4.0 | yolov5m.yaml | yolov5m.pt | yolov5m.cfg | yolov5m.weights |
yolov5l-v4.0 | yolov5l.yaml | yolov5l.pt | yolov5l.cfg | yolov5l.weights |
yolov5x-v4.0 | yolov5x.yaml | yolov5x.pt | yolov5x.cfg | yolov5x.weights |
model | .yaml | .pt | .cfg | .weights |
---|---|---|---|---|
yolov5s-v3.0 | yolov5s.yaml | yolov5s.pt | yolov5s.cfg | yolov5s.weights |
yolov5m-v3.0 | yolov5m.yaml | yolov5m.pt | yolov5m.cfg | yolov5m.weights |
yolov5l-v3.0 | yolov5l.yaml | yolov5l.pt | yolov5l.cfg | yolov5l.weights |
yolov5x-v3.0 | yolov5x.yaml | yolov5x.pt | yolov5x.cfg | yolov5x.weights |
model | .yaml | .pt | .cfg | .weights |
---|---|---|---|---|
yolov5s-v2.0 | yolov5s.yaml | yolov5s.pt | yolov5s.cfg | yolov5s.weights |
yolov5m-v2.0 | yolov5m.yaml | yolov5m.pt | yolov5m.cfg | yolov5m.weights |
yolov5l-v2.0 | yolov5l.yaml | yolov5l.pt | yolov5l.cfg | yolov5l.weights |
yolov5x-v2.0 | yolov5x.yaml | yolov5x.pt | yolov5x.cfg | yolov5x.weights |
build and run
reference the readme.md
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。