caffe实战之训练并测试自己的数据

本文详细介绍了如何在Caffe平台上利用mstar数据集训练卷积神经网络模型,包括数据准备、制作LMDB数据、训练网络以及进行测试的全过程。关键步骤包括创建数据目录结构,转换数据为LMDB格式,调整网络配置文件和训练脚本,最后进行模型测试。
摘要由CSDN通过智能技术生成

训练之前,确认caffe已经编译成功。本文以mstar数据库为例,介绍如何在caffe平台上训练卷及神经网络模型并进行测试。

一、准备数据

这个步骤是最繁琐,也最容易出错的一步,任何差池都会导致最终训练效果不如人意,

建议多花时间检查这部分,确保数据集的质量。

1、在data文件夹下新建文件夹mstar,并进入该目录,分别简历train和val文件夹。

2、在train和val文件夹下分别建立n个文件夹,n代表目标的类别数目。比如我要对mstar中10类目标进行分类,则分别简历10个文件夹,命名为0~9。

3、在各个文件夹下储存各个类别的目标图片,包括train和val。注意图片的命名,一定不要有重复。为方便,可以在windows下用看图王等批处理软件,对命名进行修改。例如可以改为’120100.jpg‘等,表示该图片为验证集第3类(存在第0类)的第100张图片。

4、在mstar文件夹下新建train.txt和val.txt,内容如下,表示图片及其对应的类别。这部分是挺繁琐的,可以在网上搜索批处理的方法,例如filepath2traintxt.sh以及filepath2valtxt.sh。

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值