anomaly-detection-api / configs /patchcore_transformers.yaml
Senum2001
Deploy Anomaly Detection API
9cf599c
raw
history blame contribute delete
817 Bytes
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