Spaces:
Build error
Build error
# Performance Benchmarking tools | |
These bash scripts will assist in measuring the training performance of the anomalib library. | |
The python script (`benchmark.py`) will assist in computing metrics for all the models in the repository. | |
## Usage | |
Run the train.sh with the same args as the tools/train.py. Refer to [`../README.md`](https://github.com/openvinotoolkit/anomalib/blob/development/README.md) for those details. | |
Note: To collect memory read/write numbers, run the script with sudo privileges. Otherwise, those values will be blank. | |
``` | |
sudo -E ./train.sh # Train STFPM on MVTec AD leather | |
sudo -E ./train.sh --config <path/to/model/config.yaml> | |
sudo -E ./train.sh --model stfpm | |
``` | |
The training script will create an output directory in this location, and inside it will have a time stamped directory for each training run you do. You can find the raw logs in there, as well as any errors captured in the train.log file. | |
For post processing, run the post-process.sh script with the results directory you want to post process. | |
``` | |
./post-process.sh ./output/2021Aug31_2351 | |
``` | |
--- | |
To use the python script, run it from the root directory. | |
``` | |
python tools/benchmarking/benchmark.py | |
``` | |
The output will be generated in results folder and a csv file for each model. | |