Commit 3c0e7e23 authored by kiegeland's avatar kiegeland
Browse files

added multilabel retrain.py and retrain.sh

parent 25e639de

cnn/multilabel/retrain.py

0 → 100755
+1067 −0

File added.

Preview size limit exceeded, changes collapsed.

cnn/multilabel/retrain.sh

0 → 100755
+12 −0
Original line number Diff line number Diff line 
python retrain.py \

--bottleneck_dir=bottlenecks \

--how_many_training_steps 50000 \

--model_dir=model_dir \

--output_graph=retrained_graph.pb \

--output_labels=retrained_labels.txt \

--summaries_dir=retrain_logs \

--flip_left_right=True \

--random_scale=20 \

--random_brightness=20 \

--random_crop=20 \

--image_dir=/softpro/ss18/caption/cocoapi/multilabel_outer/ 
 No newline at end of file