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

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

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

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


深度学习基础-数据处理入门:Caffe-TF简单的数据制作工具_第1张图片
深度学习基础-数据处理入门:Caffe-TF简单的数据制作工具_第2张图片

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


深度学习基础-数据处理入门:Caffe-TF简单的数据制作工具_第3张图片
深度学习基础-数据处理入门:Caffe-TF简单的数据制作工具_第4张图片

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


欢迎加入PyTorch交流群:637638632

欢迎关注深度学习公众号:深度智心

深度学习基础-数据处理入门:Caffe-TF简单的数据制作工具_第5张图片

你可能感兴趣的:(深度学习基础-数据处理入门:Caffe-TF简单的数据制作工具)