diff --git a/README.md b/README.md index 1e96e9b..6a5281e 100644 --- a/README.md +++ b/README.md @@ -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. diff --git a/retinaface/retina_mnet.cpp b/retinaface/retina_mnet.cpp index 121a5fb..28fc92d 100644 --- a/retinaface/retina_mnet.cpp +++ b/retinaface/retina_mnet.cpp @@ -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; } diff --git a/retinaface/retina_r50.cpp b/retinaface/retina_r50.cpp index 7629de5..c84f965 100644 --- a/retinaface/retina_r50.cpp +++ b/retinaface/retina_r50.cpp @@ -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; }