微软caffe-SSD的训练和预测（windows cpu）

下载ssd:

git clone https://github.com/conner99/caffe.git

cd caffe

git checkout ssd-microsoft

修改CommonSettings.props

在caffe-master\src\caffe下新建3rdparty文件夹，里面添加hungarian.cpp

在caffe-master\include\caffe下新建3rdparty文件夹，里面添加hungarian.hpp

下载地址：链接: http://pan.baidu.com/s/1o7MTRIi 密码: xyye

编译即可

下载数据

http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar

http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar

http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar

在data\VOC0712下解压VOC文件，变成VOC2007和VOC2012两个文件夹

在caffe根目录下新建get_image_size的bat文件

运行bat文件，生成test_name_size.txt文件

复制data\VOC0712的create_data.bat到caffe根目录下，修改如下root_dir，运行bat文件，在data\VOC0712下生成trainval_lmdb和test_lmdb文件

下载VGG的weight文件https://gist.github.com/weiliu89/2ed6e13bfd5b57cf81d6

下载设置好的训练和测试的prototxt文件（见上面百度云地址，且这里都已经设置好是CPU了，但是要注意下里面的一些文件路径与你的路径是否一致），也可以修改models\VGGNet\VOC0712下的ssd_pascal.py和score_ssd_pascal.py来生成相应的训练和测试网络文件及是否采用GPU计算等参数的设置。

在根目录下新建bat文件

进行训练，前面的CNN部分是之间用VGG的权重的。

因为只有CPU，所以训练了一晚上，迭代了2000次，测试下效果

其中test,txt代表是图像路径及名称（类似于HDF5那种形式）具体如下：（最好写绝对路径，否则可能读不到）

原来的项目是预测输出的是

代表的是类置信得分坐标值

为了在opencv显示显示其预测结果和保存结果，给ssd_detect.cpp添加一些操作（主要是对image type做了处理，video就没处理了）

预测的效果如下：

置信度阀值取0.3还能识别两类，因为训练的太少了，效果不好，所以有GPU的可以迭代个5万次左右应该就可以得到很好的结果。实验室太穷，没GPU只有训练下意思意思

微软caffe-SSD的训练和预测（windows cpu）相关推荐

笔记：caffe ssd gpu训练自己的数据集
caffe ssd GPU训练自己的数据集一.caffe环境搭建二.数据集三.训练和测试一.caffe环境搭建环境搭建过程网上有很多教程,可能要踩些坑吧,正常,这里不多赘述,主要是我当时也没 ...
笔记：caffe安装编译，并用ssd gpu训练自己的数据集
caffe ssd GPU训练自己的数据集一.caffe环境搭建二.数据集三.训练和测试一.caffe环境搭建环境搭建过程网上有很多教程,可能要踩些坑吧,正常,这里不多赘述,主要是我当时也没 ...
Caffe SSD Ubuntu16 04 训练自己的数据集
总的来说,Caffe 是一个比较难上手的框架.这次尝试训练 Caffe 框架下 SSD 模型的训练是我第一次使用 Caffe 框架.下面就说一说我踩过的几个坑,希望能够帮助到大家. 1 编译 Caff ...
(caffe入门)windows caffe 之 mnist 训练
文章目录 1. mnist 数据集下载 2. mnist 数据集转换 3. 修改网络结构文件 lenet_train_test.prototxt 和网络求解文件 lenet_solver.proto ...
微软服务器系统ssd,在UCS B200 M5上安装带M.2 SSD的操作系统(VMware、Windows)
简介本文档介绍在UCS B200 M5上安装带有M.2 SSD的操作系统(VMware.Windows) Cisco UCSB200 M5刀片服务器有一个迷你存储模块选项,可插入主板插座,以提供额外 ...
pytorch建立mobilenetV3-ssd网络并进行训练与预测
pytorch建立mobilenetV3-ssd网络并进行训练与预测前言 Step1:搭建mobilenetV3-ssd网络框架需要提前准备的函数和类. mobilenetV3_large 调用m ...
[2] SSD配置+训练VOC0712+训练自己的数据集
GitHub https://github.com/weiliu89/caffe/tree/ssd http://blog.csdn.net/u010733679/article/details/52 ...
SSD框架训练自己的数据集
SSD demo中详细介绍了如何在VOC数据集上使用SSD进行物体检测的训练和验证.本文介绍如何使用SSD实现对自己数据集的训练和验证过程,内容包括: 1 数据集的标注2 数据集的转换3 使用SSD如 ...
SSD配置+训练VOC0712+训练自己的数据集
GitHub https://github.com/weiliu89/caffe/tree/ssd http://blog.csdn.net/u010733679/article/details/52 ...
把树分成森林 matlab,20170106RF_Matlab 随机森林指的是利用多棵树对样本进行训练并预测的一种分类器，包括两个方面：数据的随 269万源代码下载- www.pudn.com...
文件名称: 20170106RF_Matlab下载收藏√ [ 5 4 3 2 1 ] 开发工具: matlab 文件大小: 441 KB 上传时间: 2017-01-06 下载次数: 0 ...

微软caffe-SSD的训练和预测（windows cpu）

微软caffe-SSD的训练和预测（windows cpu）相关推荐

最新文章

热门文章