Update retina_r50.cpp and retina_nmt.cpp (wang-xinyu#546)

lijie2160 · May 16, 2021 · 20fce2b · 20fce2b
1 parent 6a74b88
commit 20fce2b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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.