Add Caffe model support to start_profiler.sh

Change-Id: I7e15adb651f7ad9065e2d9e11c89c888ab272fff
Signed-off-by: Inki Dae <inki.dae@samsung.com>

diff --git a/start_profiler.sh b/start_profiler.sh
index e4c8103..9daa935 100644 (file)
--- a/start_profiler.sh
+++ b/start_profiler.sh
@@ -1,20 +1,30 @@
 #!/bin/sh
 
-CNT=`/usr/bin/inference_engine_test --gtest_list_tests | wc -l`
-CNT=$(($CNT - 12))
+CNT=20
 
 echo "Tflite model test case count = $CNT"
 
+# TFLITE model
 LIST=$(seq 0 $CNT)
 for i in $LIST
 do
   /usr/bin/inference_engine_test --gtest_filter=Prefix/InferenceEngineTfliteTest.Inference/$i
 done
 
+# Caffe model
+CNT=8
+
+echo "Caffe model test case count = $CNT"
+
+LIST=$(seq 0 $CNT)
+for i in $LIST
+do
+  /usr/bin/inference_engine_test --gtest_filter=Prefix/InferenceEngineCaffeTest.Inference/$i
+done
+
 # If you want to add new model tests then add script for it below
 #
-# Make sure to calculate the test case count like this,
-# CNT=$(($CNT - 31)) which depends on a number of test cases above.
+# CNT=[a number of test cases]
 # LIST=$(seq 0 $CNT)
 # for i in $LIST
 # do