134 lines
4.3 KiB
Bash
Executable File
134 lines
4.3 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
if [ -z "$NETRANS_PATH" ]; then
|
|
echo "Need to set enviroment variable NETRANS_PATH"
|
|
exit 1
|
|
fi
|
|
|
|
OVXGENERATOR=$NETRANS_PATH/pnnacc
|
|
|
|
OVXGENERATOR="$OVXGENERATOR export ovxlib"
|
|
|
|
DATASET=dataset.txt
|
|
|
|
VERIFT='FLASE'
|
|
function export_network()
|
|
{
|
|
NAME=$(basename "$1")
|
|
pushd $1
|
|
|
|
QUANTIZED=$2
|
|
|
|
if [ ${QUANTIZED} = 'float' ]; then
|
|
TYPE=float;
|
|
quantization_type="none_quantized"
|
|
generate_path='./wksp/none_quantized'
|
|
elif [ ${QUANTIZED} = 'uint8' ]; then
|
|
quantization_type="asymmetric_affine"
|
|
generate_path='./wksp/asymmetric_affine'
|
|
TYPE=quantized;
|
|
elif [ ${QUANTIZED} = 'int8' ]; then
|
|
quantization_type="dynamic_fixed_point-8"
|
|
generate_path='./wksp/dynamic_fixed_point-8'
|
|
TYPE=quantized;
|
|
elif [ ${QUANTIZED} = 'int16' ]; then
|
|
quantization_type="dynamic_fixed_point-16"
|
|
generate_path='./wksp/dynamic_fixed_point-16'
|
|
TYPE=quantized;
|
|
else
|
|
echo "=========== wrong quantization_type ! ( float / uint8 / int8 / int16 )==========="
|
|
exit -1
|
|
fi
|
|
|
|
echo " ======================================================================="
|
|
echo " =========== Start Generate $NAME ovx C code with type of ${quantization_type} ==========="
|
|
echo " ======================================================================="
|
|
|
|
mkdir -p "${generate_path}"
|
|
|
|
# if want to import c code into win IDE , change --target-ide-project command-line param from 'linux64' -> 'win32'
|
|
if [ ${QUANTIZED} = 'float' ]; then
|
|
cmd="$OVXGENERATOR \
|
|
--model ${NAME}.json \
|
|
--model-data ${NAME}.data \
|
|
--model-quantize ${NAME}.quantize \
|
|
--dtype ${TYPE} \
|
|
--pack-nbg-viplite \
|
|
--model-quantize ${NAME}_${quantization_type}.quantize \
|
|
--with-input-meta ${NAME}_inputmeta.yml\
|
|
--optimize 'VIP8000NANOQI_PLUS_PID0XB1'\
|
|
--target-ide-project 'linux64' \
|
|
--viv-sdk ${NETRANS_PATH}/pnna_sdk \
|
|
--output-path ${generate_path}/${NAME}_${quantization_type}"
|
|
else
|
|
|
|
if [ -f ${NAME}_${quantization_type}.quantize ]; then
|
|
echo -e "\033[31m using ${NAME}_${quantization_type}.quantize \033[0m"
|
|
else
|
|
echo -e "\033[31m Can not find ${NAME}_${quantization_type}.quantize \033[0m"
|
|
exit -1;
|
|
fi
|
|
|
|
cmd="$OVXGENERATOR \
|
|
--model ${NAME}.json \
|
|
--model-data ${NAME}.data \
|
|
--model-quantize ${NAME}.quantize \
|
|
--dtype ${TYPE} \
|
|
--pack-nbg-viplite \
|
|
--model-quantize ${NAME}_${quantization_type}.quantize \
|
|
--with-input-meta ${NAME}_inputmeta.yml\
|
|
--optimize 'VIP8000NANOQI_PLUS_PID0XB1'\
|
|
--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
|
|
eval $cmd
|
|
|
|
# 检查是否有至少三个参数
|
|
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
|
|
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
|
|
|
|
|
|
echo " ======================================================================="
|
|
echo " =========== End Generate $NAME ovx C code with type of ${quantization_type} ==========="
|
|
echo " ======================================================================="
|
|
|
|
popd
|
|
}
|
|
|
|
if [ "$#" -lt 2 ]; then
|
|
echo "Input a network name and quantized type ( float / uint8 / int8 / int16 )"
|
|
exit -1
|
|
fi
|
|
|
|
if [ ! -e "${1%/}" ]; then
|
|
echo "Directory ${1%/} does not exist !"
|
|
exit -2
|
|
fi
|
|
|
|
echo $1,$2,$3
|
|
export_network ${1%/} ${2%/} ${3%/}
|