By default python tools/ runs STFPM model MVTec AD leather dataset.

python tools/    # Train STFPM on MVTec AD leather

Training a model on a specific dataset and category requires further configuration. Each model has its own configuration file, config.yaml, which contains data, model and training configurable parameters. To train a specific model on a specific dataset and category, the config file is to be provided:

python tools/ --config <path/to/model/config.yaml>

Alternatively, a model name could also be provided as an argument, where the scripts automatically finds the corresponding config file.

python tools/ --model stfpm

To see a list of currently supported models, refer to page: Algorithms