yolo TensorRT almost DONE

2017-08-03 15:52:08 +02:00
parent 2ef76209a1
commit 858b3501fa
6 changed files with 96 additions and 7 deletions
@@ -70,7 +70,7 @@ int checkResult(int size, value_type *data_d, value_type *correct_d, bool device
        if(fabs(data_h[i] - correct_h[i]) > 0.0001) {
            diffs += 1;
            if(diffs < 10)
-                printf("%f %f\n", data_h[i], correct_h[i]);
+                printf("%d: %f %f\n", i, data_h[i], correct_h[i]);
        }
    }