haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

yolov5 v3.1 verified ok

Browse Source
This commit is contained in:
wang-xinyu 2020-10-30 17:06:44 +08:00
parent 683991b88c
commit df45e75182
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
yolov5/README.md
View File

@ -2,8 +2,9 @@
The Pytorch implementation is [ultralytics/yolov5](https://github.com/ultralytics/yolov5).
Currently, we support yolov5 v1.0(yolov5s only), v2.0 and v3.0.
Currently, we support yolov5 v1.0(yolov5s only), v2.0, v3.0 and v3.1.
- For yolov5 v3.1, please visit [yolov5 release v3.1](https://github.com/ultralytics/yolov5/releases/tag/v3.1), and use the latest commit of this repo.
- For yolov5 v3.0, please visit [yolov5 release v3.0](https://github.com/ultralytics/yolov5/releases/tag/v3.0), and use the latest commit of this repo.
- For yolov5 v2.0, please visit [yolov5 release v2.0](https://github.com/ultralytics/yolov5/releases/tag/v2.0), and checkout commit ['7cd092d'](https://github.com/wang-xinyu/tensorrtx/commit/7cd092d38289123442157cf7defab78e816f4440) of this repo.
- For yolov5 v1.0, please visit [yolov5 release v1.0](https://github.com/ultralytics/yolov5/releases/tag/v1.0), and checkout commit ['0504551'](https://github.com/wang-xinyu/tensorrtx/commit/0504551c0b7d0bac5f998eda349810ba410715de) of this repo.
@ -12,7 +13,7 @@ Currently, we support yolov5 v1.0(yolov5s only), v2.0 and v3.0.
- Choose the model s/m/l/x by `NET` macro in yolov5.cpp
- Input shape defined in yololayer.h
- Number of classes defined in yololayer.h
- Number of classes defined in yololayer.h, **DO NOT FORGET TO ADAPT THIS, If using your own model**
- FP16/FP32 can be selected by the macro in yolov5.cpp
- GPU id can be selected by the macro in yolov5.cpp
- NMS thresh in yolov5.cpp

2
yolov5/yololayer.h
View File

@ -96,7 +96,7 @@ namespace nvinfer1
int mYoloV5NetHeight;
int mMaxOutObject;
std::vector<Yolo::YoloKernel> mYoloKernel;
void** mAnchor;
void** mAnchor;
};
class YoloPluginCreator : public IPluginCreator