未验证 提交 09772279 编写于 作者: B bigpo 提交者: GitHub

Update unpaired_dataset.py (#131)

bug fix
上级 23dcd832
......@@ -62,7 +62,7 @@ class UnpairedDataset(BaseDataset):
def __getitem__(self, idx):
if self.is_train:
img_a_path = self.data_infos_a[idx % self.size_a]['path']
idx_b = random.randint(0, self.size_b)
idx_b = random.randint(0, self.size_b - 1)
img_b_path = self.data_infos_b[idx_b]['path']
datas = dict(A_path=img_a_path, B_path=img_b_path)
else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册