Add CenterNet based on Resnet101, TensorRT not implemented.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
2019-12-18 18:59:54 +01:00
parent 44b2bce3ff
commit df888a3457
13 changed files with 773 additions and 11 deletions
@@ -89,7 +89,6 @@ int checkResult(int size, dnnType *data_d, dnnType *correct_d, bool device) {
        data_h = data_d;
        correct_h = correct_d;
    }
-
    int diffs = 0;
    for(int i=0; i<size; i++) {
        if(data_h[i] != data_h[i] || correct_h[i] != correct_h[i] || //nan control