dataset module¶
-
class
dataset.
ImageDataset
(data_txt: str, data_shape: tuple, img_channel: int, num_label: int, alphabet: str, phase: str = 'train')[源代码]¶ 基类:
mxnet.gluon.data.dataset.Dataset
-
class
dataset.
RecordDataset
(filename, data_shape: tuple, img_channel: int, num_label: int)[源代码]¶ 基类:
mxnet.gluon.data.dataset.RecordFileDataset
A dataset wrapping over a RecordIO file contraining images Each sample is an image and its corresponding label