改写VGGNet训练自己的数据集(caffe)

caffe:使用VGGNet训练自己的数据集

前言

今天跑完了样例,又用训练好的googlenet来搞了一下物体识别,接下来就得训练自己的模型了。发现许多操作跟样例是相似的,所以打算先进一步研读样例中层数的编写过程,然后再自己动手修改。

将数据集转换为lmdb格式

参考了一下人家写的

这个可以有,效率更高,暂时先不弄了,直接跑jpg格式数据集
今天还是要弄,刚开始出现错误aborted的错误,考虑可能原因
1.路径没设置好,可能是需要绝对路径,但是我却设置成了相对路径。
2.刚开始不能有train_lmdb和test_lmdb这两个文件夹,不然会冲突。

准备VGGNet16模型

这是从caffe model zoo上面下载的,这个是人家整理的,自己改改然后就能用我觉得。
现在是solver.prototxt和train_val.prototxt写完了,训练出了个错误,
在这里插入图片描述考虑可能的原因:
1.数据集没弄好,自己直接用的jpg格式,可能是train_val.prototxt里面对数据层的解释没有弄好。
2.测试集数量太少,只有一百来张,然而solver.prototxt里面的某个迭代参数等东西设置不合理。
3.自己的数据集仍然是分为十类了,是不是不应该再分子文件夹。

遇到的问题

在这里插入图片描述啊?难道是CUDA的问题吗?查看网上的说法:
在这里插入图片描述但是我运行后,
在这里插入图片描述CUDA版本是8.0啊
还有这样说的:
在这里插入图片描述然后尝试编译个新的caffe,做了一下make runtest,发现没有通过,错误如下,
在这里插入图片描述这个错误跟刚才上面训练数据时遇到的错误是一样的。

也有人这样说:
在这里插入图片描述就是有cuda7的安装残留,需要将这个去掉。

有个人这样解决了:
在这里插入图片描述这靠谱吗?好像是不行,没有安装nvidia-cuda-toolkit,没有卸载一说。

跑了一下自带的Lenet,正常的话如下:
在这里插入图片描述但我的VGGNet:
在这里插入图片描述

解决

因为我发现连make runtest都通不过,干脆解决这个问题那么无法训练的问题就解决了。我没办法了,就把caffe重新编译了一遍,哎呀,竟然通过了,然后自己又一训练,发现竟然成功了哈哈哈。
在这里插入图片描述加油啊得!

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值