train/data/null_dataset.py

from data.base_dataset import BaseDataset
import os


class NullDataset(BaseDataset):

    def __init__(self, opt):
        BaseDataset.__init__(self, opt)

    def __getitem__(self, index):
        return {'A_paths': os.path.join(self.opt.dataroot, '%d.jpg' % index)}

    def __len__(self):
        """Return the total number of images in the dataset."""
        return self.opt.max_dataset_size