-
Notifications
You must be signed in to change notification settings - Fork 54
/
setup.sh
20 lines (18 loc) · 1.2 KB
/
setup.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# Download and unzip necessary files
# 1. anno list files (Annotation copyright belongs to MS COCO team)
mkdir ./datasets
mkdir ./datasets/ms_coco
mkdir ./datasets/ms_coco/mscoco_anno_files
wget -O ./datasets/ms_coco/mscoco_anno_files/mscoco_anno_files.zip http://www.cs.jhu.edu/~jhmao/open_source_data/mrnn_cr/mscoco_anno_files.zip
unzip ./datasets/ms_coco/mscoco_anno_files/mscoco_anno_files.zip -d ./datasets/ms_coco/mscoco_anno_files/
rm ./datasets/ms_coco/mscoco_anno_files/mscoco_anno_files.zip
# 2. inception-v3 model
mkdir ./external/tf_cnn_models
wget -O ./external/tf_cnn_models/inception_dec_2015.zip https://storage.googleapis.com/download.tensorflow.org/models/inception_dec_2015.zip
unzip ./external/tf_cnn_models/inception_dec_2015.zip -d ./external/tf_cnn_models/
mv ./external/tf_cnn_models/tensorflow_inception_graph.pb ./external/tf_cnn_models/inception_v3.pb
rm ./external/tf_cnn_models/inception_dec_2015.zip
# 3. the trained mRNN model
wget -O ./trained_models/coco_caption/mrnn_GRU_570K.zip http://www.cs.jhu.edu/~jhmao/open_source_data/tf_mrnn/mrnn_GRU_570K.zip
unzip -o ./trained_models/coco_caption/mrnn_GRU_570K.zip -d ./trained_models/coco_caption/
rm ./trained_models/coco_caption/mrnn_GRU_570K.zip