待完成任务(一) 利用自己的数据集 利用lenet网络进行训练

来源:互联网 发布:万达怎么了知乎 编辑:程序博客网 时间:2024/05/18 01:06

硬件问题,使用imagenet报内存溢出,只能先用lenet进行学习,本文的主要目的是利用自己的图像库,将图库裁剪成32*32大小,然后利用lenet网络进行训练。


caffe模型输入图片大小:imagenet为256*256,cifar10为32*32,lenet为28*28


今天把数据集重新做成32*32大小,然后进行转换成lmdb文件中遭遇滑铁卢。。。。


三种模型:

imagenet      的转换脚本 是对256*256大小的,单纯改成32*32会报错,无法生成lmdb

cifar10          的转换脚本 查看其数据集是.bin(二进制形式),和图像库数据不同,可能需要转成.bin文件,                                    用 cifar的转换脚本也行不通

mnist(lenet)   的转换脚本  数据集是.ubyte,也不同,暂时也行不通


(一)imagenet的转换文件格式脚本截图



(二)cifar10文件格式查看 已经 转换文件格式脚本 截图



(二)mnist文件格式查看 已经 转换文件格式脚本 截图



问题瞬间就来了,先列在这里,待解决吧。找到方案再回来更新。

问题一:如何利用自己的数据集 利用lenet(mnist)网络进行训练

问题二:如何利用自己的数据集 利用cifar10网络进行训练

其实上面的问题关键就是如何将自己的数据集转换成相应模型需要的数据类型。

另外经过今天的学习,感觉昨天的内存溢出的错误不知道是不是搞错了,好吧,还是先继续往前看看吧,有新的见解再重新更新。


暂时搁置。



0 0
原创粉丝点击