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

fix

Browse Source
This commit is contained in:
wang-xinyu 2020-09-07 21:43:15 +08:00
parent 6603efbe16
commit 2abc7b8375
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@ -10,6 +10,7 @@ All the models are implemented in pytorch or mxnet first, and export a weights f
## News
- `7 Sep 2020`. Implement retinaface(mobilenet0.25), and got 333fps on GTX1080.
- `28 Aug 2020`. [BaofengZan](https://github.com/BaofengZan) added a tutorial for compiling and running tensorrtx on windows.
- `16 Aug 2020`. [upczww](https://github.com/upczww) added a python wrapper for yolov5.
- `14 Aug 2020`. Update yolov5 to v3.0 release.

2
retinaface/retina_mnet.cpp
View File

@ -100,7 +100,7 @@ float iou(float lbox[4], float rbox[4]) {
return interBoxS / ((lbox[2] - lbox[0]) * (lbox[3] - lbox[1]) + (rbox[2] - rbox[0]) * (rbox[3] - rbox[1]) -interBoxS + 0.000001f);
}
bool cmp(decodeplugin::Detection& a, decodeplugin::Detection& b) {
bool cmp(const decodeplugin::Detection& a, const decodeplugin::Detection& b) {
return a.class_confidence > b.class_confidence;
}

2
retinaface/retina_r50.cpp
View File

@ -98,7 +98,7 @@ float iou(float lbox[4], float rbox[4]) {
return interBoxS / ((lbox[2] - lbox[0]) * (lbox[3] - lbox[1]) + (rbox[2] - rbox[0]) * (rbox[3] - rbox[1]) -interBoxS + 0.000001f);
}
bool cmp(decodeplugin::Detection& a, decodeplugin::Detection& b) {
bool cmp(const decodeplugin::Detection& a, const decodeplugin::Detection& b) {
return a.class_confidence > b.class_confidence;
}