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

Update retina_r50.cpp and retina_nmt.cpp (#546)

Browse Source
This commit is contained in:
Huang ZhuangZhuang 2021-05-16 11:31:33 +08:00 committed by GitHub
parent 6a74b88b4e
commit 20fce2bce4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
retinaface/retina_mnet.cpp
View File

@ -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.

4
retinaface/retina_r50.cpp
View File

@ -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.