SSD踩坑记录

SSD踩坑记录

原github项目地址，借用大神的模型自己训练SSD：https://github.com/balancap/SSD-Tensorflow
1、Error：Default MaxPoolingOp only supports NHWC on device type CPU

DATA_FORMAT = 'NCHW' #修改为'NHWC'

2、关于dataset_name：
将图像数据转换为tfrecods格式时，tf_convert_data.py中：

--dataset_name=pascalvoc
--dataset_dir=./VOC2007/
--output_name=voc_2007_train
--output_dir=./tfrecords_

训练时，train_ssd_network.py中：

--train_dir=./train_model/
--dataset_dir=./tfrecords_/
--dataset_name=pascalvoc_2007
--dataset_split_name=train
--model_name=ssd_300_vgg

在自己乱改名字经过一系列报错之后发现，转换时’output_name’跟训练时pascalvoc_2007中FILE_PATTERN要对应；转换时’output_dir’跟训练时’dataset_dir’要对应。

本文观点不一定正确，仅供自己学习记录