|
1 | 1 | <div align="center">
|
2 | 2 |
|
3 |
| -<img src="docs/source/images/logos/anomalib-wide.png" width="400px"> |
| 3 | +<img src="docs/source/images/logos/anomalib-wide-blue.png" width="600px"> |
4 | 4 |
|
5 | 5 | **A library for benchmarking, developing and deploying deep learning anomaly detection algorithms**
|
6 | 6 | ___
|
@@ -69,31 +69,31 @@ python tools/train.py # Train PADIM on MVTec leather
|
69 | 69 | ```
|
70 | 70 |
|
71 | 71 | Training a model on a specific dataset and category requires further configuration. Each model has its own configuration
|
72 |
| -file, [`config.yaml`](https://gitlab-icv.inn.intel.com/algo_rnd_team/anomaly/-/blob/development/stfpm/anomalib/models/stfpm/config.yaml) |
| 72 | +file, [`config.yaml`](https://gitlab-icv.inn.intel.com/algo_rnd_team/anomaly/-/blob/development/padim/anomalib/models/padim/config.yaml) |
73 | 73 | , which contains data, model and training configurable parameters. To train a specific model on a specific dataset and
|
74 | 74 | category, the config file is to be provided:
|
75 | 75 |
|
76 | 76 | ```bash
|
77 | 77 | python tools/train.py --model_config_path <path/to/model/config.yaml>
|
78 | 78 | ```
|
79 | 79 |
|
80 |
| -For example, to train [STFPM](anomalib/models/stfpm) you can use |
| 80 | +For example, to train [PADIM](anomalib/models/padim) you can use |
81 | 81 |
|
82 | 82 | ```bash
|
83 |
| -python tools/train.py --model_config_path anomalib/models/stfpm/config.yaml |
| 83 | +python tools/train.py --model_config_path anomalib/models/padim/config.yaml |
84 | 84 | ```
|
85 | 85 |
|
86 | 86 | Alternatively, a model name could also be provided as an argument, where the scripts automatically finds the corresponding config file.
|
87 | 87 |
|
88 | 88 | ```bash
|
89 |
| -python tools/train.py --model stfpm |
| 89 | +python tools/train.py --model padim |
90 | 90 | ```
|
91 | 91 |
|
92 | 92 | where the currently available models are:
|
93 | 93 |
|
94 | 94 | - [CFlow](anomalib/models/cflow)
|
95 |
| -- [PADIM](anomalib/models/padim) |
96 | 95 | - [PatchCore](anomalib/models/patchcore)
|
| 96 | +- [PADIM](anomalib/models/padim) |
97 | 97 | - [STFPM](anomalib/models/stfpm)
|
98 | 98 | - [DFM](anomalib/models/dfm)
|
99 | 99 | - [DFKDE](anomalib/models/dfkde)
|
|
0 commit comments