add profile para

netrans_cli/export.sh

@@ -81,41 +81,37 @@ function export_network()
             --target-ide-project 'linux64' \
             --viv-sdk ${NETRANS_PATH}/pnna_sdk \
             --output-path ${generate_path}/${NAME}_${quantization_type}"   
-    fi  
-    if [${VERIFY}='TRUE']; then
-        echo $cmd
-    fi
+    fi 
+
+    # if [ "${VERIFY}"='TRUE' ]; then
+    #     echo $cmd
+    # fi
     eval $cmd
 
-    # copy input file into source code folder
-    # sourcefile="`cat ${DATASET}`"
-    # cpcmd="cp -fr $sourcefile ${generate_path}/"
-    
-    # echo $cpcmd
-    # eval $cpcmd
- 
-    # temp='wksp/temp'
-
-    # mkcmd="mkdir -p ${temp}"
-    # eval $mkcmd
-
-    # sourcefile="`cat ${DATASET}`"
-    # cpcmd="cp -fr $sourcefile ${temp}/"
-    
-    # echo $cpcmd
-    # eval $cpcmd
-    
-    cpcmd="cp ${generate_path}_nbg_viplite/network_binary.nb ${generate_path}/"
-    eval $cpcmd
+    # 检查是否有至少三个参数
+    if [ $# -ge 3 ]; then
+        # 检查第三个参数是否为 "profile"
+        if [ "$3" == "profile" ]; then
+            cpcmd="cp ${generate_path}_nbg_viplite/network_binary.nb ${generate_path}/"
+            eval $cpcmd
      
-    delcmd="rm -rf ${generate_path}_nbg_viplite"
-    eval $delcmd
+            delcmd="rm -rf ${generate_path}_nbg_viplite"
+            eval $delcmd
+        fi
+    else 
+        # mvcmd="mv ${generate_path}_nbg_viplite ${generate_path}"
+        # eval $mvcmd
+        tmp='wksp/tmp'
+        mkdir -p ${tmp}
+        cpcmd="cp ${generate_path}_nbg_viplite/network_binary.nb ${tmp}/"
+        eval $cpcmd
+        
+        delcmd="rm -rf ${generate_path}  ${generate_path}_nbg_viplite"
+        eval $delcmd
+        mv ${tmp} ${generate_path}
+    fi
 
-    # rm -rf ${generate_path}
-    # mvcmd="mv ${temp} ${generate_path}"
 
-    # eval $mvcmd
-    
     echo " ======================================================================="
     echo " =========== End  Generate $NAME ovx C code with type of ${quantization_type} ==========="
     echo " ======================================================================="
@@ -133,4 +129,5 @@ if [ ! -e "${1%/}" ]; then
     exit -2
 fi
 
-export_network ${1%/} ${2%/}
+echo $1,$2,$3
+export_network ${1%/} ${2%/} ${3%/}