From 20fce2bce47168bfa936930e15d094498cb18eda Mon Sep 17 00:00:00 2001 From: Huang ZhuangZhuang Date: Sun, 16 May 2021 11:31:33 +0800 Subject: [PATCH] Update retina_r50.cpp and retina_nmt.cpp (#546) --- retinaface/retina_mnet.cpp | 4 ++-- retinaface/retina_r50.cpp | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/retinaface/retina_mnet.cpp b/retinaface/retina_mnet.cpp index 8555df5..bef3f39 100644 --- a/retinaface/retina_mnet.cpp +++ b/retinaface/retina_mnet.cpp @@ -12,8 +12,8 @@ #define USE_FP16 // set USE_INT8 or USE_FP16 or USE_FP32 #define DEVICE 0 // GPU id #define BATCH_SIZE 1 -#define CONF_THRESH = 0.75 -#define IOU_THRESH = 0.4 +#define CONF_THRESH 0.75 +#define IOU_THRESH 0.4 // stuff we know about the network and the input/output blobs static const int INPUT_H = decodeplugin::INPUT_H; // H, W must be able to be divided by 32. diff --git a/retinaface/retina_r50.cpp b/retinaface/retina_r50.cpp index d2e51d1..f5c4413 100644 --- a/retinaface/retina_r50.cpp +++ b/retinaface/retina_r50.cpp @@ -12,8 +12,8 @@ #define USE_INT8 // set USE_INT8 or USE_FP16 or USE_FP32 #define DEVICE 0 // GPU id #define BATCH_SIZE 1 -#define CONF_THRESH = 0.75 -#define IOU_THRESH = 0.4 +#define CONF_THRESH 0.75 +#define IOU_THRESH 0.4 // stuff we know about the network and the input/output blobs static const int INPUT_H = decodeplugin::INPUT_H; // H, W must be able to be divided by 32.