深度学习中，数据是基础。对于自己的数据，为了方便的应用开源库Caffe和tensorflow，可以将数据转成其需要的格式，比如LMDB和tfrecords。当然还有很多其他的方式，比如对于Caffe可以直接读取图像格式数据，hdf5格式等，甚至可以自己写一个python的DataLayer，但有时候由于各种原因，仍然需要我们将数据转化为LMDB格式。
今天分享一段很简单的数据转换代码，如下：
首先我们先写这个数据转换类：
然后我们就可以调用这个类来处理数据了，以下生成一些假数据，做个测试：
抱歉简书上传代码不太方便，粘贴过来格式很难看，大家就自己写一遍，更加有成就感。
欢迎加入PyTorch交流群：637638632
欢迎关注深度学习公众号：深度智心
网友评论