budui/raycv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

v2

Browse Source
This commit is contained in:
budui
2020-10-22 22:42:01 +08:00
parent 0019d4034c
commit 376f5caeb7
11 changed files with 140 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
data/dataset/image_translation.py
View File

@@ -38,9 +38,9 @@ class SingleFolderDataset(Dataset):
@DATASET.register_module()
class GenerationUnpairedDataset(Dataset):
def __init__(self, root_a, root_b, random_pair, pipeline, with_path=False):
self.A = SingleFolderDataset(root_a, pipeline, with_path)
self.B = SingleFolderDataset(root_b, pipeline, with_path)
def __init__(self, root_a, root_b, random_pair, pipeline_a, pipeline_b, with_path=False):
self.A = SingleFolderDataset(root_a, pipeline_a, with_path)
self.B = SingleFolderDataset(root_b, pipeline_b, with_path)
self.with_path = with_path
self.random_pair = random_pair