用自己的数据集做resnet-50的测试

本文档记录了使用9分类图片数据集进行ResNet-50模型训练的过程,包括数据预处理、模型调整以及遇到的问题与解决方案。主要问题包括:尺寸适配、标签转换、loss不变、读取图片错误、caffe训练报错以及精度指标的解释。
摘要由CSDN通过智能技术生成

这是我参考的博客

我的数据集是一个图片9分类的数据集,文本标签的形式为路径+空格+标签,制作的标签方法参考我上一篇博客

做一些问题补充如下:

1、因为我做的图片数据集稳128*128,所以把prototxt最后pool5里的kernelsize改小了一些,然后9分类所以把最后的numout改成了9

2、文本标签我做的是1从9出现了错误,我把它改成了0-8后解决

3、出现loss一直不变的问题,基于官方训练好后的50caffemodel进行训练得以解决

4、出现了cv2_image_data无法读取图片文件的报错,确认路径是好的,最终查找到原因是text路径文件多出一个空行造成

5、caffe-parallel has a bug: Check failed: proto.SerializeToOstream(&output)训练的时候出现这个错误

网上有说是没有相应的目录或者权限对这个目录进行写操作而引起的。
修改权限:cd /User/Yourname/caffe/example/myfile #你的caffe目录下的训练文件目录

sudo chmod 777 ./* #修改权限

或者是在solver中设置错了snapshot directory 

但我的原因是内存不够导致!!!!

 

6、

layer {
  name: "accuracy-1"
  type: "Accuracy"
  bot
  • 2
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值