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