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 8 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
8 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
f708abf
Merge main and resolve conflicts
samet-akcay Mar 26, 2025
ee9d0a9
Merge main
samet-akcay Mar 26, 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
58 changes: 38 additions & 20 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,19 @@
# Datamodules
from .datamodules.base import AnomalibDataModule
from .datamodules.depth import DepthDataFormat, Folder3D, MVTec3D
from .datamodules.image import VAD, BTech, Datumaro, Folder, ImageDataFormat, Kolektor, MVTec, MVTecAD, MVTecLOCO, Visa
from .datamodules.image import (
VAD,
BTech,
Datumaro,
Folder,
ImageDataFormat,
Kolektor,
MVTec,
MVTecAD,
MVTecLOCO,
RealIAD,
Visa,
)
from .datamodules.video import Avenue, ShanghaiTech, UCSDped, VideoDataFormat

# Datasets
Expand Down Expand Up @@ -146,40 +158,46 @@ 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", # Include MVTec for backward compatibility
"MVTec",
"MVTecAD",
"MVTecADDataset",
"MVTecLOCO",
"MVTecLOCODataset",
"RealIAD",
"VAD",
"VADDataset",
"Visa",
"VisaDataset",
# Video
# Video Data Modules
"Avenue",
"AvenueDataset",
"ShanghaiTech",
"ShanghaiTechDataset",
"UCSDped",
# Datasets
"Folder3DDataset",
"MVTec3DDataset",
"BTechDataset",
"DatumaroDataset",
"FolderDataset",
"KolektorDataset",
"MVTecADDataset",
"MVTecLOCODataset",
"VADDataset",
"VisaDataset",
"AvenueDataset",
"ShanghaiTechDataset",
"UCSDpedDataset",
"VideoDataFormat",
# Predict
"PredictDataset",
# Functions
"get_datamodule",
# Exceptions
"UnknownDatamoduleError",
]
6 changes: 5 additions & 1 deletion src/anomalib/data/datamodules/image/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@
from .kolektor import Kolektor
from .mvtec_loco import MVTecLOCO
from .mvtecad import MVTec, MVTecAD
from .realiad import RealIAD
from .vad import VAD
from .visa import Visa

Expand All @@ -50,6 +51,7 @@ class ImageDataFormat(str, Enum):
- ``MVTEC_AD``: MVTec AD Dataset
- ``MVTEC_3D``: MVTec 3D AD Dataset
- ``MVTEC_LOCO``: MVTec LOCO Dataset
- ``REALIAD``: Real-IAD Dataset
- ``VAD``: Valeo Anomaly Detection Dataset
- ``VISA``: Visual Anomaly Dataset
"""
Expand All @@ -62,6 +64,7 @@ class ImageDataFormat(str, Enum):
MVTEC_AD = "mvtecad"
MVTEC_3D = "mvtec_3d"
MVTEC_LOCO = "mvtec_loco"
REAL_IAD = "realiad"
VAD = "vad"
VISA = "visa"

Expand All @@ -71,9 +74,10 @@ class ImageDataFormat(str, Enum):
"Datumaro",
"Folder",
"Kolektor",
"MVTecAD",
"MVTec", # Include both for backward compatibility
"MVTecAD",
"MVTecLOCO",
"RealIAD",
"VAD",
"Visa",
]
Loading