Centernet: fix pooling problem, add centrnet demo

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com> Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2020-01-21 12:50:18 +01:00
parent c68d6f318e
commit 7f239efdc0
12 changed files with 229 additions and 935 deletions
@@ -55,8 +55,8 @@ Pooling::Pooling( Network *net, int winH, int winW, int strideH, int strideW,
    int padW = paddingW == 0? winW -1 : paddingW;
    
    if(final){
-        h = (h + padH - winH)/strideH +1 +1;
-        w =  (w + padW - winW)/strideW +1 +1;
+        h = (h + 2*paddingH - winH)/strideH +1 ;
+        w =  (w + 2*paddingW - winW)/strideW +1;
    }
    else{
        h = (h + padH - winH)/strideH +1;