WinCLIP с нулевым выстрелом из anomalib не работает ⇐ Python

[Anonymous]

После недавних достижений в области классификации/сегментации аномалий я захотел попробовать новую модель winCLIP, реализация которой также есть в библиотеке anomalib.
Как протестировать нулевой или нулевой выстрел дать пару "нормальных/здоровых" снимков для малокадра? Поскольку это модель с нулевым выстрелом, я предполагаю, что она легко заработает из коробки, но я не могу заставить ее работать. Вот мой текущий код:
from anomalib.models.image import WinClip
from anomalib.engine import Engine

# Import the datamodule
from anomalib.data import Folder
from anomalib.data.utils import TestSplitMode

# Create the datamodule
datamodule = Folder(
name="lasercut_plank",
root="./DATA_0shot",
normal_dir="normal",
test_split_mode=TestSplitMode.NONE
)

# Setup the datamodule
datamodule.setup()

# Access the datasets
train_dataset = datamodule.train_data

# Access the dataloaders
train_dataloader = datamodule.train_dataloader()

# Create the model and engine
model = WinClip(class_name="lasercut_plank")
engine = Engine(task="segmentation")

# Train a Patchcore model on the given datamodule
engine.train(datamodule=datamodule, model=model)


Подробнее здесь: https://stackoverflow.com/questions/792 ... ot-working