MXNet源码解读：数据读取基础类—mxnet.io.DataIter

最新推荐文章于 2020-09-18 11:48:08 发布

AI之路

最新推荐文章于 2020-09-18 11:48:08 发布

阅读量3.3k

点赞数 5

分类专栏： MXNet 深度学习 MXNet使用及源码解读

本文链接：https://blog.csdn.net/u014380165/article/details/79720131

版权

MXNet的mxnet.io.DataIter是数据迭代器基础，用于图像算法等数据读取接口。理解该类有助于掌握MXNet数据处理，并能自定义数据封装。源码显示next方法用于获取batch数据，结束时抛出StopIteration。

摘要由CSDN通过智能技术生成

mxnet.io.DataIter是MXNet框架中构造数据迭代器的基础类，在MXNet框架下只要和数据读取相关的接口基本上都继承该类，比如我们常用的图像算法相关的mxnet.io.ImageRecordIter类或mxnet.image.ImageIter类都直接或间接继承mxnet.io.DataIter类进行封装。因此熟悉mxnet.io.DataIter类对于其他high level的数据读取接口以及自己封装数据非常有帮助。

mxnet.io.DataIter类的源码地址

源码如下。从注释可以看出该类是MXNet框架下构造数据迭代器的基础类，比较抽象。在该类中主要通过调用next方法返回下一个batch的数据，如果数据已迭代读取完毕，则抛出StopIteration。

class DataIter(object):
    """The base class for an MXNet data iterator.

    All I/O in MXNet is handled by specializations of this class. Data iterators
    in MXNet are similar to standard-iterators in Python. On each call to `next`
    they return a `DataB