File size: 1,620 Bytes
7fe0374 |
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 47 48 49 50 51 52 53 54 |
_target_: src.data.anomaly_clip_datamodule.AnomalyCLIPDataModule
data_dir:
train: /home/hzx/Projects/Data/Visa
valid: /home/hzx/Projects/Data/MVTec-AD
test: /home/hzx/Projects/Data/MVTec-AD
dataset:
train:
_target_: src.data.components.anomal_dataset.VisaDataset
_partial_: true
transform:
_target_: src.data.components.transform.ImageTransform
image_size: 336
mask_transform:
_target_: src.data.components.transform.MaskTransform
image_size: ${data.image_size}
preload: false
valid:
_target_: src.data.components.anomal_dataset.MVTecDataset
_partial_: true
transform:
_target_: src.data.components.transform.ImageTransform
image_size: 336
mask_transform:
_target_: src.data.components.transform.MaskTransform
image_size: ${data.image_size}
preload: false
aug_rate: 0.0
test:
_target_: src.data.components.anomal_dataset.MVTecDataset
_partial_: true
transform:
_target_: src.data.components.transform.ImageTransform
image_size: 336
mask_transform:
_target_: src.data.components.transform.MaskTransform
image_size: ${data.image_size}
preload: false
aug_rate: 0.0
kshot:
_target_: src.data.components.kshot_dataset.MVTecKShotDataset
_partial_: true
k_shot: 1
transform:
_target_: src.data.components.transform.ImageTransform
image_size: 336
mask_transform:
_target_: src.data.components.transform.MaskTransform
image_size: ${data.image_size}
preload: false
image_size: 336
num_workers: 6
pin_memory: False
batch_size: 8
|