SSD-Tensorflow-master----傻瓜3步走-让你的GPU训练跑起来

注意：默认读者已经配置好tensorflow-gpu及相关显卡训练环境

一、下载SSD-Tensorflow-master和数据集VCO

下载链接

• SSD-Tensorflow-master: https://codeload.github.com/balancap/SSD-Tensorflow/zip/master
• VOC
  • https://pjreddie.com/media/files/VOCtrainval_11-May-2012.tar
  • https://pjreddie.com/media/files/VOCtrainval_06-Nov-2007.tar
  • https://pjreddie.com/media/files/VOCtest_06-Nov-2007.tar

二、修改配置

• SSD-Tensorflow-master>>tf_convert_data.py

  
  改：34行名字可以随意取,但要与53行的一致修改，这里我改为‘pascalvoc_2007’
  
  改：37行,改为你解压训练数据集的路径,即它 VOCtrainval_06-Nov-2007.tar 解压的voc2007
  
  改：40行转换格式后的每一个文本名，
  
  注意：40行名字 前缀 要与SSD-Tensorflow-master>>datasets>>pascalvoc2007.py里的 ‘voc_2007_’ 一致，这里我在前缀基础上+‘train‘
  ’
  
  43行 装换格式后的存放的文件夹路径，这里我新建了一个 'train_tfrecords’文件夹
  

• SSD-Tensorflow>>datasets>>pascalvoc_to_tfrecords.py

  改：83行里的 ’r‘ 改为 ’rb‘
  

• SSD-Tensorflow-master>>train_ssd_network.py

  改：133行，转换格式的数据集名字，之前我命名的是 ’passcalvoc_2007‘
  
  改：139行，这里是之前转换格式文件存放的文件夹路径，我是放在 ’train_tfrecords‘ 里
  
  改：113 行，原 ’0.01‘ 改为—> ‘0.001’ ,这里不改也能跑，但是损失值会一直是 ’nan‘
  

三、执行两个文件

• 在SSD-Tensorflow-master下打开终端

• 执行

  python tf_convert_data.py

  完成train_voc2007数据集格式转换

• 执行

  python train_ssd_network.py

  训练开始

测试 SSD-Tensorflow-master 两步走