SSD-Tensorflow 512x512 訓練配置
搞了幾天終於把這個給搞得差不多了,遇到的錯誤這裡也記錄一下:
一、配置【配置什麼的300和512其實差不多,這裡只舉一個例子來分析一下】
之前的檔案修改什麼的和300x300的一樣:https://www.cnblogs.com/GrPhoenix/p/10018072.html
從自己訓練的ssd_300_vgg模型開始訓練ssd_512_vgg的模型
因ssd_300_vgg中沒有block12,又因為block7,block8,block9,block10,block11,中的引數張量兩個網路模型中不匹配,因此ssd_512_vgg中這幾個模組的引數不從ssd_300_vgg模型中繼承,因此使用checkpoint_exclude_scopes命令指出。
因為所有的引數均需要訓練,因此不使用命令--trainable_scopes
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
1
#/bin/bash
2
DATASET_DIR
=
/
home
/
data
/
xxx
/
imagedata
/
xing_tf /
train_tf
/
3
TRAIN_DIR
=
/
home
/
data
/
xxx
/
model
/
xing300512_model
/
4
CHECKPOINT_PATH
=
/
home
/
data
/
xxx
/
model
/
xing300_model
/
model.ckpt
-
60000
#載入的ssd_300_vgg模型
5
python3 .
/
train_ssd_network.py \
6
-
-
train_dir
=
${TRAIN_DIR} \
7
-
-
dataset_dir
=
${DATASET_DIR} \
8
-
-
dataset_name
=
pascalvoc_2007 \
9
-
-
dataset_split_name
=
train \
10
-
-
model_name
=
ssd_512_vgg \
11
-
-
checkpoint_path
=
${CHECKPOINT_PATH} \
12
-
-
checkpoint_model_scope
=
ssd_300_vgg \
13
-
-
checkpoint_exclude_scopes
=
ssd_512_vgg
/
block7,ssd_512_vgg
/
block7_box,ssd_512_vgg
/
block8,ssd_512_vgg
/
block8_box,ssd_512_vgg
/
block9,ssd_512_vgg
/
block9_box,ssd_512_vgg
/
block10,ssd_512_vgg
/
block10_box,ssd_512_vgg
/
block11,ssd_512_vgg
/
b lock11_box,ssd_512_vgg
/
block12,ssd_512_vgg
/
block12_box \
14
#--trainable_scopes=ssd_300_vgg/conv6,ssd_300_vgg/conv7,ssd_300_vgg/block8,ssd_300_vgg/block9,ssd_300_vgg/block10,ssd_300_vgg/block11,ssd_300_vgg/block4_box,ssd_300_vgg/block7_box,ssd_300_vgg/block8_box,ssd_300_vgg/block9_box,ssd_3 00_vgg/block10_box,ssd_300_vgg/block11_box \
15
-
-
save_summaries_secs
=
28800
\
16
-
-
save_interval_secs
|