一、 配置修改
-
SSD-Tensorflow-master>>tf_convert_data.py
改:要转换的数据集名称:34行名字可以随意取,但要与53行的一致修改,这里我改为‘pascalvoc_2007’
改:37行,改为你解压测试数据集的路径,即它 VOCtestval_06-Nov-2007.tar 解压的voc2007
改:40行转换格式后的每一个文本名,
注意:40行名字 前缀 要与SSD-Tensorflow-master>>datasets>>pascalvoc2007.py里的 ‘voc_2007_’ 一致,这里我在前缀基础上+‘test‘
’
43行 装换格式后的存放的文件夹路径,这里我新建了一个 'tests_tfrecords’文件夹
-
SSD-Tensorflow-master>>eval_ssd_network.py
改:75行,已训练好的模型存放的路径,我的是放在 ‘checkpoints’ 里
改:79行,测试结果存放路径,我把它跟模型放一块了 ‘checkpoints’
改:139行,这里是之前转换格式文件存放的文件夹路径,我是放在 ’tests_tfrecords‘ 里
改:90行,核心架构名称,‘ssd_300_vgg‘,这个跟训练的文件默认的一样
订正:上边eval_ssd_network.py有一行一个地方忘了改了
改:84行,这里与转格式时tf_convert_data.py里的34行名字一致
二、执行两个文件
-
在SSD-Tensorflow-master下打开终端
-
执行
python tf_convert_data.py
完成train_voc2007数据集格式转换
-
执行
python eval_ssd_network.py
测试开始