training model on custom dataset

[blackdevil112]

Your Question

HI
can you mention the steps to train these models on custom dataset

we would appreciate for immediate response

Thanks

Forum Check

• I have searched the discussions forum for an answer to my question.

[github-actions[bot]]

Thanks for submitting this issue! It has been added to our triage queue. A maintainer will review it shortly.

[rajeshgangireddy]

Hi @blackdevil112
I am assuming you have a dataset and want to train anomalib models on it.
A simple example for that is shown below :

from anomalib.data import Folder
from anomalib.models import Patchcore
from anomalib.engine import Engine


dataset_folder = "path/to/dataset"

datamodule = Folder(
    name="DatasetName",  # A unique name for your dataset
    root=dataset_folder,  # Root directory containing 'normal' and 'abnormal' folders
    normal_dir="train/normal",  # Name of the folder containing normal images
    abnormal_dir = "test/abnormal",
    normal_test_dir = "test/normal",
    # explore other inputs 
    train_batch_size=32,
    eval_batch_size=32,
    num_workers=8,
)

datamodule.setup()

model = Patchcore()
engine = Engine()
engine.fit(datamodule=datamodule, model=model)