20180815

dataset_structure

58266	dog
58432	not_dog

input_params

python retrain_lsvrc_dataset.py \
--tfhub_module=https://tfhub.dev/google/imagenet/mobilenet_v1_100_224/feature_vector/1 \
--image_dir=/home/ubuntu/datasets/pps_imagenet/output_dnd \
--output_graph=/home/ubuntu/datasets/pps_output/output_graph.pb \
--output_labels=/home/ubuntu/datasets/pps_output/output_labels.txt \
--adam_optimizer_epsilon=0.1 \
--how_many_training_steps=36500 \
--learning_rate_decay_steps=3650

bazel run --config=opt \
//tensorflow/contrib/lite/toco:toco -- \
--input_file=/home/ubuntu/datasets/pps_output/output_graph.pb \
--output_file=/home/ubuntu/datasets/pps_output/output_graph.tflite \
--inference_type=FLOAT \
--input_shape=1,224,224,3 \
--input_array=Placeholder \
--output_array=final_result \
--input_format=TENSORFLOW_GRAPHDEF \
--output_format=TFLITE