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
@@ -45,7 +45,7 @@ struct threshold : public thrust::binary_function<float,float,float>
 {
  __host__ __device__
  float operator()(float x, float y) { 
-    float toll = 1e-6; 
+    double toll = 1e-6; 
    if(fabsf(x-y)>toll) 
        return 0.0f; 
    else