Tensorflow workspaces for training your custom models of K210
mobilenetv1/models/mobilenet_v1.py
, ATTENTION, K210 does not support the method of SAME PADDING in tensorflow, so we need to add padding around the image manually before a stride=2
conv (in this situation, the padding method of conv layer with stride=2
should be set to VALID
mobilenetv1/run_mobilenet_v1.sh
and start your training.freeze_graph.py
to freeze your model from ckpt
to pb
, just run python mobilenetv1/freeze_graph.py model.mobilenet_v1 ckpt_fold pb_file
python mobilenetv1/validation_imagenet.py pb_file val_set_fold
python mobilenetv1/predict_one_pic.py pb_file pic