|
@@ -109,7 +109,9 @@ def build_dataset(is_train, config): |
|
|
# dataset_train = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'images'), txt_path=os.path.join(config.DATA.DATA_PATH, 'images', 'images.txt'), transform=transform) |
|
|
# dataset_train = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'images'), txt_path=os.path.join(config.DATA.DATA_PATH, 'images', 'images.txt'), transform=transform) |
|
|
# dataset_val = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'val'), txt_path=os.path.join(config.DATA.DATA_PATH, 'val', 'val.txt'), transform=transform) |
|
|
# dataset_val = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'val'), txt_path=os.path.join(config.DATA.DATA_PATH, 'val', 'val.txt'), transform=transform) |
|
|
# dataset = torch.utils.data.ConcatDataset([dataset_train, dataset_val]) |
|
|
# dataset = torch.utils.data.ConcatDataset([dataset_train, dataset_val]) |
|
|
dataset = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'train'), txt_path=os.path.join(config.DATA.DATA_PATH, 'p2_train.txt'), transform=transform) |
|
|
|
|
|
|
|
|
dataset_train = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'train'), txt_path=os.path.join(config.DATA.DATA_PATH, 'p2_train.txt'), transform=transform) |
|
|
|
|
|
dataset_val = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'test'), txt_path=os.path.join(config.DATA.DATA_PATH, 'p2_val.txt'), transform=transform) |
|
|
|
|
|
dataset = torch.utils.data.ConcatDataset([dataset_train, dataset_val]) |
|
|
else: |
|
|
else: |
|
|
# dataset = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'val'), txt_path=os.path.join(config.DATA.DATA_PATH, 'val', 'val.txt'), transform=transform) |
|
|
# dataset = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'val'), txt_path=os.path.join(config.DATA.DATA_PATH, 'val', 'val.txt'), transform=transform) |
|
|
dataset = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'test'), txt_path=os.path.join(config.DATA.DATA_PATH, 'p2_val.txt'), transform=transform) |
|
|
dataset = ImageDataset(root=os.path.join(config.DATA.DATA_PATH, 'test'), txt_path=os.path.join(config.DATA.DATA_PATH, 'p2_val.txt'), transform=transform) |
|
|