runs STFPM model
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