Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

🚀 Add Real-IAD dataset #2545

Open
wants to merge 6 commits into
base: main
Choose a base branch
from
Open

🚀 Add Real-IAD dataset #2545

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
c2844e8
Add Real-IAD dataset implementation
samet-akcay Feb 3, 2025
3d5b7b7
merge main
samet-akcay Mar 6, 2025
4fb5d8e
license update
samet-akcay Mar 6, 2025
c80c7ab
Add generation of dummy realiad dataset for tests
samet-akcay Mar 7, 2025
9680990
remove redundant checks
samet-akcay Mar 7, 2025
705ddd2
Revert the latest commit
samet-akcay Mar 7, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 14 additions & 0 deletions examples/configs/data/realiad.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
class_path: anomalib.data.RealIAD
init_args:
root: ./datasets/Real-IAD
category: audiojack
resolution: 256
train_batch_size: 32
eval_batch_size: 32
num_workers: 8
test_split_mode: none
val_split_mode: same_as_test
train_augmentations: null
val_augmentations: null
test_augmentations: null
augmentations: null
41 changes: 24 additions & 17 deletions src/anomalib/data/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@
# Datamodules
from .datamodules.base import AnomalibDataModule
from .datamodules.depth import DepthDataFormat, Folder3D, MVTec3D
from .datamodules.image import BTech, Datumaro, Folder, ImageDataFormat, Kolektor, MVTecAD, Visa
from .datamodules.image import BTech, Datumaro, Folder, ImageDataFormat, Kolektor, MVTec, MVTecAD, RealIAD, Visa
from .datamodules.video import Avenue, ShanghaiTech, UCSDped, VideoDataFormat

# Datasets
Expand Down Expand Up @@ -144,35 +144,42 @@ def get_datamodule(config: DictConfig | ListConfig | dict) -> AnomalibDataModule
"NumpyVideoItem",
"VideoBatch",
"VideoItem",
# Depth
# Data Formats
"DataFormat",
"DepthDataFormat",
"ImageDataFormat",
"VideoDataFormat",
# Depth Data Modules
"Folder3D",
"Folder3DDataset",
"MVTec3D",
"MVTec3DDataset",
# Image
# Image Data Modules
"BTech",
"BTechDataset",
"Datumaro",
"DatumaroDataset",
"Folder",
"FolderDataset",
"ImageDataFormat",
"Kolektor",
"KolektorDataset",
"MVTec",
"MVTecAD",
"MVTecADDataset",
"RealIAD",
"Visa",
"VisaDataset",
# Video
# Video Data Modules
"Avenue",
"AvenueDataset",
"ShanghaiTech",
"ShanghaiTechDataset",
"UCSDped",
# Datasets
"Folder3DDataset",
"MVTec3DDataset",
"BTechDataset",
"DatumaroDataset",
"FolderDataset",
"KolektorDataset",
"MVTecADDataset",
"VisaDataset",
"AvenueDataset",
"ShanghaiTechDataset",
"UCSDpedDataset",
"VideoDataFormat",
# Predict
"PredictDataset",
# Functions
"get_datamodule",
# Exceptions
"UnknownDatamoduleError",
]
4 changes: 4 additions & 0 deletions src/anomalib/data/datamodules/image/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@
from .folder import Folder
from .kolektor import Kolektor
from .mvtecad import MVTec, MVTecAD
from .realiad import RealIAD
from .visa import Visa


Expand All @@ -45,6 +46,7 @@ class ImageDataFormat(str, Enum):
- ``KOLEKTOR``: Kolektor Surface-Defect Dataset
- ``MVTEC_AD``: MVTec AD Dataset
- ``MVTEC_3D``: MVTec 3D AD Dataset
- ``REALIAD``: Real-IAD Dataset
- ``VISA``: Visual Inspection for Steel Anomaly Dataset
"""

Expand All @@ -55,6 +57,7 @@ class ImageDataFormat(str, Enum):
KOLEKTOR = "kolektor"
MVTEC_AD = "mvtecad"
MVTEC_3D = "mvtec_3d"
REAL_IAD = "realiad"
VISA = "visa"


Expand All @@ -65,5 +68,6 @@ class ImageDataFormat(str, Enum):
"Kolektor",
"MVTecAD",
"MVTec", # Include both for backward compatibility
"RealIAD",
"Visa",
]
Loading