import mxnet as mx def load_data_rec(self, batch_size): train_iter = mx.image.ImageIter( path_imgrec=self._data_dir_ + "torcs_train.rec", data_shape=(3, 210, 280), # (channels, height, width) batch_size=batch_size, label_width=14, data_name='image', label_name='predictions_label' ) test_iter = mx.image.ImageIter( path_imgrec=self._data_dir_ + "torcs_test.rec", data_shape=(3, 210, 280), # (channels, height, width) batch_size=batch_size, label_width=14, data_name='image', label_name='predictions_label' ) data_mean = None data_std = None return train_iter, test_iter, data_mean, data_std