Spaces:
Sleeping
Sleeping
File size: 817 Bytes
9cf599c | 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 | model:
class_path: anomalib.models.Patchcore
init_args:
backbone: "wide_resnet50_2"
layers: ["layer2", "layer3"]
coreset_sampling_ratio: 0.1
num_neighbors: 9
data:
# You can also use: class_path: anomalib.data.Folder (re-export)
class_path: anomalib.data.datamodules.image.folder.Folder
init_args:
name: "transformers"
root: "./dataset"
normal_dir: "train/normal" # ONLY normal images here
abnormal_dir: "test/faulty" # faulty images for eval
normal_test_dir: "test/normal" # a few normals for eval
train_batch_size: 4
eval_batch_size: 4
num_workers: 4
# (optional) add augmentations later via train_augmentations/val_augmentations/test_augmentations
trainer:
max_epochs: 1
accelerator: "auto"
devices: 1
enable_checkpointing: true
|