深度学习基础-数据处理入门:Caffe-TF简单的数据制作工具

深度学习中,数据是基础。对于自己的数据,为了方便的应用开源库Caffe和tensorflow,可以将数据转成其需要的格式,比如LMDB和tfrecords。当然还有很多其他的方式,比如对于Caffe可以直接读取图像格式数据,hdf5格式等,甚至可以自己写一个python的DataLayer,但有时候由于各种原因,仍然需要我们将数据转化为LMDB格式。

今天分享一段很简单的数据转换代码,如下:

首先我们先写这个数据转换类:


然后我们就可以调用这个类来处理数据了,以下生成一些假数据,做个测试:


抱歉简书上传代码不太方便,粘贴过来格式很难看,大家就自己写一遍,更加有成就感。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容