Yolov3_tiny works on tensorRT

Signed-off-by: fbagni <gattinomicino>
2019-12-23 16:11:41 +01:00
parent f0ea2c027f
commit 7e7f480e2b
5 changed files with 20 additions and 32 deletions
@@ -17,7 +17,7 @@ __global__ void resize_kernel(  int i_N,float *x, int i_w, int i_h, int i_c,
    int out_c = i%o_c;
    i = i/o_c;

-    //copying last column/last row
+    //copying last column/last row as padding
    int in_index = ((i*i_c + MIN(out_c,i_c-1))*i_h +  MIN(out_h,i_h-1))*i_w + MIN(out_w, i_w-1);
    out[out_index] = x[in_index];
 }
@@ -43,4 +43,4 @@ void resizeForward(  dnnType* srcData, dnnType* dstData, int n, int i_c, int i_h
        // printDeviceVector(i_size, srcData);
        // printDeviceVector(o_size, dstData);
    }
-}
+}