Skip to content

Instantly share code, notes, and snippets.

@universax
Last active February 14, 2019 02:40
Show Gist options
  • Save universax/970176b5b32fe73c935d30054a3a64a4 to your computer and use it in GitHub Desktop.
Save universax/970176b5b32fe73c935d30054a3a64a4 to your computer and use it in GitHub Desktop.
Transfer learning from pre trained model
# tensorflow/models/research/slim
# datasets needs to be converted to tfrecord format.
python train_image_classifier.py \
--train_dir "/home/user_name/train" \
--dataset_dir "/home/user_name/dataset" \
--model_name "mobilenet_v1" \
--dataset_name=mymodel \
--dataset_split_name=train \
--checkpoint_path "/home/user_name/mobilenet/mobilenet_v1_1.0_224.ckpt" \
--train_image_size=224 \
--max_number_of_steps=1000 \
--quantize=True \
--checkpoint_exclude_scopes=MobilenetV1/Logits
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment