Skip to content

Commit 91653ca

Browse files
authored
🐞 Fix MVTecLOCO args (open-edge-platform#2650)
* Fix MVTecLOCO args Signed-off-by: Samet Akcay <[email protected]> * Revert test/val split ratios Signed-off-by: Samet Akcay <[email protected]> --------- Signed-off-by: Samet Akcay <[email protected]>
1 parent 161bb36 commit 91653ca

File tree

2 files changed

+12
-11
lines changed

2 files changed

+12
-11
lines changed

src/anomalib/data/datamodules/image/mvtec_loco.py

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -77,14 +77,15 @@ class MVTecLOCO(AnomalibDataModule):
7777
Defaults to ``None``.
7878
augmentations (Transform | None): General augmentations to apply if stage-specific
7979
augmentations are not provided.
80-
test_split_mode (TestSplitMode): Method to create test set.
80+
Defaults to ``None``.
81+
test_split_mode (TestSplitMode | str): Method to create test set.
8182
Defaults to ``TestSplitMode.FROM_DIR``.
82-
test_split_ratio (float): Fraction of data to use for testing.
83-
Defaults to ``0.2``.
84-
val_split_mode (ValSplitMode): Method to create validation set.
85-
Defaults to ``ValSplitMode.SAME_AS_TEST``.
86-
val_split_ratio (float): Fraction of data to use for validation.
87-
Defaults to ``0.5``.
83+
val_split_mode (ValSplitMode | str): Method to create validation set.
84+
Defaults to ``ValSplitMode.FROM_DIR``.
85+
test_split_ratio (float | None): Fraction of data to use for testing.
86+
Defaults to ``None``.
87+
val_split_ratio (float | None): Fraction of data to use for validation.
88+
Defaults to ``None``.
8889
seed (int | None, optional): Seed for reproducibility.
8990
Defaults to ``None``.
9091
@@ -130,9 +131,9 @@ def __init__(
130131
val_augmentations: Transform | None = None,
131132
test_augmentations: Transform | None = None,
132133
augmentations: Transform | None = None,
133-
test_split_mode: TestSplitMode | str | None = None,
134+
test_split_mode: TestSplitMode | str = TestSplitMode.FROM_DIR,
135+
val_split_mode: ValSplitMode | str = ValSplitMode.FROM_DIR,
134136
test_split_ratio: float | None = None,
135-
val_split_mode: ValSplitMode | str | None = None,
136137
val_split_ratio: float | None = None,
137138
seed: int | None = None,
138139
) -> None:
@@ -145,8 +146,8 @@ def __init__(
145146
test_augmentations=test_augmentations,
146147
augmentations=augmentations,
147148
test_split_mode=test_split_mode,
148-
test_split_ratio=test_split_ratio,
149149
val_split_mode=val_split_mode,
150+
test_split_ratio=test_split_ratio,
150151
val_split_ratio=val_split_ratio,
151152
seed=seed,
152153
)

src/anomalib/utils/path.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -327,7 +327,7 @@ def generate_output_filename(
327327
category_index = input_path.parts.index(category, dataset_index)
328328
start_index = category_index + 1
329329
except ValueError:
330-
msg = f"Category '{category}' not found in the input path after the dataset name."
330+
msg = f"Category '{category}' not found in the input path after the dataset name {dataset_name}."
331331
raise ValueError(msg) from None
332332

333333
# Preserve all subdirectories after the category (or dataset if no category)

0 commit comments

Comments
 (0)