A question about Datasets

[ztk985]

Hello author, thank you for providing the project code. But I would like to ask you, how should I prepare the data set before training the model

[roberto722]

Up

[jamdodot]

same question
self.target need to be set, such as bottle?

[TooTouch]

@ztk985 @roberto722

I'm sorry too late.
You can download MVTecAD datasets in here.

[TooTouch]

@jamdodot

Yes.
If you want to train a model for bottle class, you need to set the target as bottle.

python main.py --yaml_config ./configs.yaml DATASET.target capsule

[jamdodot]

@jamdodot

Yes. If you want to train a model for bottle class, you need to set the target as bottle.

python main.py --yaml_config ./configs.yaml DATASET.target capsule

I also want to ask [example] model overview.ipynb. Is this file the inference process? Is it using a trained model?

[TooTouch]

@jamdodot

[example] model overview.ipynb is a tutorial example for one batch of training dataset.
The purpose of the notebook file is to check the process of the MemSeg.

[jamdodot]

Hello, sorry to bother you. How can i set up training and testing on other datasets. such as KolektorSDD & ELPV