NN: Add shell script to convert models.

This commit is contained in:
iabdalkader 2018-06-16 18:52:25 +02:00
parent e4144184d1
commit 5be410c139

9
ml/cmsisnn/nn_run_all.sh Executable file
View File

@ -0,0 +1,9 @@
#!/usr/bin/env sh
if [ -z $1 ]; then
echo "Usage : nn_run_all.sh model_name"
exit 1
fi
MODEL=${1}
set -v
python2 nn_quantizer.py --model models/${MODEL}/${MODEL}_train_test.prototxt --weights models/${MODEL}/${MODEL}_iter_*.caffemodel --save models/${MODEL}/${MODEL}.pkl --gpu
python2 nn_convert.py --model models/${MODEL}/${MODEL}.pkl --mean caffe/examples/${MODEL}/mean.binaryproto --output models/${MODEL}/${MODEL}.network