Merge remote-tracking branch 'origin/master' into cnet

scripts/test_all_tests.sh

@@ -72,11 +72,18 @@ do
     # ./test_imuodom &>> $out_file
     # print_output $? imuodom
 
+    test_net shelfnet
+    test_net shelfnet_berkeley
     test_net yolo4
+    test_net yolo4_320
+    test_net yolo4_320_coco2
+    test_net yolo4_512
+    test_net yolo4_608
     test_net yolo4-csp
     test_net yolo4x
     test_net yolo4_berkeley
     test_net yolo4tiny
+    test_net yolo4tiny_512
     test_net yolo3
     test_net yolo3_berkeley
     test_net yolo3_coco4

scripts/test_inference.sh

@@ -0,0 +1,52 @@
+#!/bin/bash
+
+function test_inference {
+    ./test_$1
+    ./test_rtinference $1_$2.rt 1 
+    ./test_rtinference $1_$2.rt 4
+}
+
+sudo jeston_clock
+
+# modes=( 1 ) # only FP32
+# modes=( 1 2 ) # FP32 and FP16
+modes=( 1 2 3 ) # FP32, FP16 and INT8
+
+rm times_rtinference.csv
+for i in "${modes[@]}"
+do
+    rm *rt
+    if [ $i -eq 1 ]
+    then
+        export TKDNN_MODE=FP32
+        mode=fp32
+        echo -e "${ORANGE}Test FP32${NC}"
+    fi
+    if [ $i -eq 2 ]
+    then
+        export TKDNN_MODE=FP16
+        mode=fp16
+        echo -e "${ORANGE}Test FP16${NC}"
+    fi
+    if [ $i -eq 3 ]
+    then
+        export TKDNN_MODE=INT8
+        export TKDNN_CALIB_LABEL_PATH=../demo/COCO_val2017/all_labels.txt
+        export TKDNN_CALIB_IMG_PATH=../demo/COCO_val2017/all_images.txt
+        mode=int8
+        echo -e "${ORANGE}Test INT8${NC}"
+        
+	fi
+
+    export TKDNN_BATCHSIZE=4
+    echo -e "${ORANGE}Batch $TKDNN_BATCHSIZE ${NC}"
+    
+    test_inference yolo4_320 $mode
+    test_inference yolo4 $mode
+    test_inference yolo4_512 $mode
+    test_inference yolo4_608 $mode
+    test_inference yolo4tiny $mode
+done
+
+
+