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