windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告

本人花了好几天,顺利完成windows版本caffe-ssd和ubuntu版本的caffe-ssd环境配置、脚本修改、最终独创一套高速自定义训练的项目包,可以在配置caffe ssd环境后可以迅速训练自己的数据,而不需要眼花缭乱的搞各种脚本,进行各种转换,这么麻烦了。虽然谈不上是专家,但是已经比较熟了,里面很多脚本都是本人亲自操刀书写。先来一张windows下面训练自己定义的数据窗口显示

windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告

再来一张ubuntu下面CPU版本训练的caffe ssd截图

windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告

两个平台的都差不多但是脚本有点区别,windows主要使用bat文件,而ubunutu主要是用python和shell脚本,使用方法我分为2个平台介绍,在windows下面,只要放到指定位置,基本不用改代码,这是windows上面好处

实验环境:

Windows10专业版64位

Caffe-ssd CPU版本

实验要求:

源码必须放在D:\caffe-ssd-microsoft,然后将myproj放进caffe-ssd-microsoft文件夹

windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告

 

 

(1)打开myproj文件夹进去data文件夹里面的

D:\caffe-ssd-microsoft\myproj\data\VOCdevkit\VOC2012然后JPEGImages放进去自己的jpg文件,Annotations放进自己的标注文件

  1. 删除D:\caffe-ssd-microsoft\myproj/lmdb里面的train_lmdb文件夹和test_lmdb文件夹,不然生成lmdb会提示错误
  2. bat文件夹里面有个xml2txt.py,这个是将图片转为txt文件,dos窗口切换到bat目录,执行python xml2txt.py,这时候txt文件夹会生成对应的train.txt和test.txt,里面的内容是图片文件名 xml文件名这样的格式,注意这个脚本是将数据集80%作为训练集,20为测试集,可以再脚本里面改
  3. bat文件夹里面有个get_name_size.py,dos窗口切换到bat目录,执行python get_name_size.py,这时候txt文件夹会生成对应的test_name_size.txt,里面的内容是文件名无后缀 高 宽这样的格式
  4. 双击bat文件夹里面create_lmdb.bat即可生成train_lmdb和test_lmdb,注意双击前确保train_lmdb和test_lmdb文件夹不要存在,否则会报错
  5. 双击bat文件夹里面的starttrain.bat即可开始训练、

在ubunutu平台,由于都是使用的绝对路径需要改自己的用户名,其他基本不用改

实验环境:

Ubuntu16.04

Caffe-ssd CPU版本

实验要求:

新建一个/home/caffe-ssd目录,将caffe放进去,然后将myproj文件夹和caffe放同级目录

windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告

windows 10+ubuntu16.04全程配置CPU版本的caffe ssd并训练自己的数据研究报告

 

 

(1)打开myproj文件夹进去data文件夹里面的

/home/fut/caffe-ssd/myproj/data/VOCdevkit/VOC2012

然后JPEGImages放进去自己的jpg文件,Annotations放进自己的标注文件

  1. 删除/home/fut/caffe-ssd/myproj/lmdb里面的train_lmdb文件夹和test_lmdb文件夹,不然生成lmdb会提示错误
  2. 打开xml2txt.py修改里面用户名,我的是fut改改用户名就可以了,用替换的方法就可以,然后终端切换到script目录,执行python xml2txt.py,这时候txt文件夹会生成对应的train.txt和test.txt,里面的内容是图片文件名 xml文件名这样的格式,注意这个脚本是将数据集80%作为训练集,20为测试集,可以再脚本里面改。如果你遇到caffe ssd安装配置问题,可以私信我,有偿帮忙。
  3. 打开get_name_size.py修改里面用户名,我的是fut改改用户名就可以了,用替换的方法就可以,然后终端切换到script目录,执行python get_name_size.py,这时候txt文件夹会生成对应的test_name_size.txt,里面的内容是文件名无后缀 高 宽这样的格式
  4. 终端切换到script目录,执行python train_lmdb_gen.py即可生成lmdb,生成的文件在lmdb里面查看
  5. 终端切换到script目录,执行python test_lmdb_gen.py即可生成lmdb,生成的文件在lmdb里面查看
  6. 终端切换到script目录,执行python ssd_pascal.py即可开始训练,其中很多参数ssd_pascal.py可以修改,这里只需要把用户名换成自己的就可以了,这个是做检测用的。

 

由于手头没有gpu电脑所以没有配置GPU版本,但是这些都是小CASE。