Training

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

python tools/train.py    # 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/train.py --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/train.py --model stfpm

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