diff options
| author | Jun-Yan Zhu <junyanz@users.noreply.github.com> | 2017-08-27 14:39:36 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-08-27 14:39:36 -0700 |
| commit | 60e4af620c93ffd1f056a1e161ce1c73e8d0fddd (patch) | |
| tree | 487bd2eca1019c93e64a408dabcb7f0971c0c545 | |
| parent | 174682626f21b4222a1cb294348e59ad3b260eb7 (diff) | |
Update unaligned_dataset.py
| -rw-r--r-- | data/unaligned_dataset.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/data/unaligned_dataset.py b/data/unaligned_dataset.py index b162869..d31eb05 100644 --- a/data/unaligned_dataset.py +++ b/data/unaligned_dataset.py @@ -26,7 +26,7 @@ class UnalignedDataset(BaseDataset): def __getitem__(self, index): A_path = self.A_paths[index % self.A_size] index_A = index % self.A_size - index_B = random.randint(0, self.B_size) + index_B = random.randint(0, self.B_size - 1) B_path = self.B_paths[index_B] # print('(A, B) = (%d, %d)' % (index_A, index_B)) A_img = Image.open(A_path).convert('RGB') |