mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
NN: Add shell script to convert models.
This commit is contained in:
parent
e4144184d1
commit
5be410c139
9
ml/cmsisnn/nn_run_all.sh
Executable file
9
ml/cmsisnn/nn_run_all.sh
Executable 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
|
||||
Loading…
Reference in New Issue
Block a user