-
Notifications
You must be signed in to change notification settings - Fork 20
Expand file tree
/
Copy pathretrain.sh
More file actions
17 lines (16 loc) · 725 Bytes
/
retrain.sh
File metadata and controls
17 lines (16 loc) · 725 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
OUTPUT_DIRECTORY=$HOME/storage/PAN/cnn/
TRAIN_DIR=$HOME/storage/PAN/cnn/train_dir/
VALIDATION_DIR=$HOME/storage/PAN/cnn/valid_dir/
LABELS_FILE=$HOME/storage/PAN/cnn/labels.txt
INCEPTION_MODEL_DIR=$HOME/storage/PAN/cnn/fish/model/inception-v3
MODEL_PATH="${INCEPTION_MODEL_DIR}/model.ckpt-157585"
FISH_DATA_DIR=$HOME/storage/PAN/cnn/fish
TRAIN_DIR=$HOME/storage/PAN/cnn/fish/save
cd ~/storage/PAN/tensorflow/
python tensorflow/examples/image_retraining/retrain.py \
--bottleneck_dir=$FISH_DATA_DIR/bottlenecks \
--how_many_training_steps 500 \
--model_dir=$FISH_DATA_DIR/model \
--output_graph=$FISH_DATA_DIR/moel/retrained_graph.pb \
--output_labels=$FISH_DATA_DIR/retrained_labels.txt \
--image_dir /tf_files/flower_photos