Skip to content

Commit 6d4ce00

Browse files
Add AnomalyDINO to model list. Also alphabetically re-order some models
1 parent ef05fb3 commit 6d4ce00

File tree

2 files changed

+8
-3
lines changed

2 files changed

+8
-3
lines changed

src/anomalib/models/__init__.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@
5555
from anomalib.utils.path import convert_snake_to_pascal_case, convert_to_snake_case, convert_to_title_case
5656

5757
from .image import (
58+
AnomalyDINO,
5859
Cfa,
5960
Cflow,
6061
Csflow,
@@ -93,6 +94,8 @@ class UnknownModelError(ModuleNotFoundError):
9394

9495

9596
__all__ = [
97+
"AiVad",
98+
"AnomalyDINO",
9699
"Cfa",
97100
"Cflow",
98101
"Csflow",
@@ -104,6 +107,7 @@ class UnknownModelError(ModuleNotFoundError):
104107
"EfficientAd",
105108
"Fastflow",
106109
"Fre",
110+
"Fuvas",
107111
"Ganomaly",
108112
"Padim",
109113
"Patchcore",
@@ -114,8 +118,6 @@ class UnknownModelError(ModuleNotFoundError):
114118
"UniNet",
115119
"VlmAd",
116120
"WinClip",
117-
"AiVad",
118-
"Fuvas",
119121
]
120122

121123
logger = logging.getLogger(__name__)

src/anomalib/models/image/__init__.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
>>> predictions = engine.predict(model=model, datamodule=datamodule) # doctest: +SKIP
2424
2525
Available Models:
26+
- :class: `AnomalyDINO`: Boost Memorybank Models with DINOv2
2627
- :class:`Cfa`: Contrastive Feature Aggregation
2728
- :class:`Cflow`: Conditional Normalizing Flow
2829
- :class:`Csflow`: Conditional Split Flow
@@ -44,6 +45,7 @@
4445
- :class:`WinClip`: Zero-/Few-Shot CLIP-based Detection
4546
"""
4647

48+
from .anomaly_dino import AnomalyDINO
4749
from .cfa import Cfa
4850
from .cflow import Cflow
4951
from .csflow import Csflow
@@ -67,10 +69,12 @@
6769
from .winclip import WinClip
6870

6971
__all__ = [
72+
"AnomalyDINO",
7073
"Cfa",
7174
"Cflow",
7275
"Csflow",
7376
"Dfkde",
77+
"Dinomaly",
7478
"Dfm",
7579
"Draem",
7680
"Dsr",
@@ -87,5 +91,4 @@
8791
"UniNet",
8892
"VlmAd",
8993
"WinClip",
90-
"Dinomaly",
9194
]

0 commit comments

Comments
 (0)