File size: 1,155 Bytes
061b20f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
dataset_config = dict(
type = 'DSIFN',
data_root = 'data/DSIFN',
train_mode = dict(
transform = dict(
# RandomSizeAndCrop = {"size": 512, "crop_nopad": False},
RandomHorizontallyFlip = {"p": 0.5},
RandomVerticalFlip = {"p": 0.5},
RandomGaussianBlur = None,
),
loader = dict(
batch_size = 4,
num_workers = 4,
pin_memory=True,
shuffle = True,
drop_last = True
),
),
val_mode = dict(
transform = dict(
# RandomSizeAndCrop = {"size": 512, "crop_nopad": False},
),
loader = dict(
batch_size = 4,
num_workers = 4,
pin_memory=True,
shuffle = False,
drop_last = False
)
),
test_mode = dict(
transform = dict(
# RandomSizeAndCrop = {"size": 512, "crop_nopad": False},
),
loader = dict(
batch_size = 4,
num_workers = 4,
pin_memory=True,
shuffle = False,
drop_last = False
)
),
)
|