Change CenterNet input dimension.

This commit changes the image input dimension, it updates the
CenterNet detection class.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
Davide Sapienza
2020-02-06 18:08:06 +01:00
parent c695d8c5d7
commit 70eb5214cc
5 changed files with 22 additions and 14 deletions
+3 -3
View File
@@ -73,14 +73,14 @@ class CenternetDetection {
//processing
float toll = 0.000001;
int K = 100;
int width = 56; // TODO
int width = 128;//56; // TODO
public:
dnnType *rt_out[4];
float inp_height = 224;//512;
float inp_width = 224;//512;
float inp_height = 512;//224;//512;
float inp_width = 512;//224;//512;
int classes = 80;
int num = 0;