caffe ssd样本训练的流程

本文介绍如何在Ubuntu 16.04环境下配置SSD目标检测框架,包括数据集准备、环境搭建步骤、lmdb文件生成及模型训练过程。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

配置好环境的ubuntu16.04.

训练的样本为jpg和xml文件。

在/home/user下的data中创建项目名称的目录,其下可按日期创建目录,如20180718。在其中包含三个文件夹Annotations  ImageSets JPEGImages。xml文件放在Annotations中,jpg放在JPEGImages中。

xml和jpg文件一一对应。可编写提取文件名到文本文件中的脚本,将样本的名字提取到train.txt中,抽取其中部分名字到val.txt中,其之和等于原始样本的总数。更改后文件名数目为train:val = 5 :1左右(我最近的一次样本训练是8000:1000+)。

将train.txt和val.txt文件一起放到ImageSets中的Main目录中。

执行caffe ssd对应的文件夹中的caffe/data/项目名中的create_list.sh和create_data.sh文件。

生成对应可用于样本训练的lmdb文件。

将lmdb文件拷入用于样本训练的models中的项目名中,其中包含:example目录,lmdb, label**.prototxt, solver_test.prototxt和solver_train.prototxt,还有train的sh文件。要在example中的test和train的prototxt更改lmdb的路径。执行train.sh。

生成对应的.caffemodel文件。将此文件放入caffe ssd对应文件的models中的项目下,此时在项目名中包含如下文件

若缺少lmdb,要补上。

修改merge_bn.py中的相应名称和路径。再对其进行执行,可得到example中两个deploy文件。

完成。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值