From 461a2e878c236814626858a9326c9863fc197b26 Mon Sep 17 00:00:00 2001 From: pengyc Date: Fri, 19 May 2023 20:35:34 +0800 Subject: [PATCH] change the dataset --- GAN/pi-GAN/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GAN/pi-GAN/datasets.py b/GAN/pi-GAN/datasets.py index 0da126d..7a1702f 100644 --- a/GAN/pi-GAN/datasets.py +++ b/GAN/pi-GAN/datasets.py @@ -80,7 +80,7 @@ def __init__(self, dataset_path, img_size, **kwargs): super().__init__() paths = glob.glob(dataset_path) self.data = [] - for path in paths[0:3]: + for path in paths: self.data = self.data + glob.glob(path + '/rgb/*.png') assert len(self.data) > 0, "Can't find data; make sure you specify the path to your dataset" self.transform = transforms.Compose(