/
run2.sh
53 lines (40 loc) · 2.54 KB
/
run2.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
#!/bin/bash
# 查找脚本所在路径,并进入
#DIR="$( cd "$( dirname "$0" )" && pwd )"
DIR=$PWD
cd $DIR
echo current dir is $PWD
# 设置目录,避免module找不到的问题
export PYTHONPATH=$PYTHONPATH:$DIR/im2txt
# 定义各目录
output_dir=/output # 训练目录
dataset_dir=/data/HataFeng/data-flickr8-wordcounts # 数据集目录,这里是写死的,记得修改
dataset_dir_train=/data/HataFeng/data-flickr8/train-?????-of-00008
dataset_dir_ckpt=/data/HataFeng/data-msco/inception_v3.ckpt
dataset_dir_val=/data/HataFeng/data-flickr8-val/val-?????-of-00001
dataset_dir_vocab_file=/data/HataFeng/data-flickr8-wordcounts/word_counts.txt
dataset_dir_infe_file1=/data/HataFeng/data-flickr8-wordcounts/Filkr8_1928319708_ccf1f4ee72.jpg
dataset_dir_infe_file2=/data/HataFeng/data-flickr8-wordcounts/COCO_val2014_000000224477.jpg
dataset_dir_infe_file3=/data/HataFeng/data-flickr8-wordcounts/Random_1.jpg
train_dir=$output_dir/train
checkpoint_dir=$train_dir
eval_dir=$output_dir/eval
# config文件
#config=ssd_mobilenet_v1_pets.config
#pipeline_config_path=$output_dir/$config
# 因为dataset里面的东西是不允许修改的,所以这里要把config文件复制一份到输出目录
#cp $DIR/object_detection/samples/configs/$config $pipeline_config_path
echo "############ training with V3 #################"
python ./im2txt/train.py --input_file_pattern=$dataset_dir_train --inception_checkpoint_file=$dataset_dir_ckpt --train_dir=$train_dir --train_inception=false --number_of_steps=70001
echo "############ inference #################"
# 在test.jpg上验证导出的模型
python ./im2txt/run_inference.py --checkpoint_path=$checkpoint_dir --vocab_file=$dataset_dir_vocab_file --input_files=$dataset_dir_infe_file1
#echo "############ inference #################"
#python ./im2txt/run_inference.py --checkpoint_path=$checkpoint_dir --vocab_file=$dataset_dir_vocab_file --input_files=$dataset_dir_infe_file2
#echo "############ inference #################"
#python ./im2txt/run_inference.py --checkpoint_path=$checkpoint_dir --vocab_file=$dataset_dir_vocab_file --input_files=$dataset_dir_infe_file3
#echo "############ evaluate #################"
#python im2txt/evaluate.py --input_file_pattern=$dataset_dir_val --checkpoint_dir=$checkpoint_dir --eval_dir=$eval_dir
echo "############ training for Fine Tune #################"
python ./im2txt/train.py --input_file_pattern=$dataset_dir_train --inception_checkpoint_file=$checkpoint_dir --train_dir=$train_dir --train_inception=true --number_of_steps=310001
echo "############ end #################"