Add transform config to yaml fil (#156)

openvinotoolkit · Mar 23, 2022 · 2dfa0a7 · 2dfa0a7
1 parent b57e025
commit 2dfa0a7
Show file tree

Hide file tree

Showing 8 changed files with 39 additions and 28 deletions.
diff --git a/anomalib/data/__init__.py b/anomalib/data/__init__.py
@@ -45,6 +45,8 @@ def get_datamodule(config: Union[DictConfig, ListConfig]) -> LightningDataModule
             test_batch_size=config.dataset.test_batch_size,
             num_workers=config.dataset.num_workers,
             seed=config.project.seed,
+            transform_config=config.dataset.transform_config,
+            create_validation_set=config.dataset.create_validation_set,
         )
     elif config.dataset.name.lower() == "btech":
         datamodule = BTechDataModule(
@@ -56,6 +58,8 @@ def get_datamodule(config: Union[DictConfig, ListConfig]) -> LightningDataModule
             test_batch_size=config.dataset.test_batch_size,
             num_workers=config.dataset.num_workers,
             seed=config.project.seed,
+            transform_config=config.dataset.transform_config,
+            create_validation_set=config.dataset.create_validation_set,
         )
     else:
         raise ValueError(

diff --git a/anomalib/models/cflow/config.yaml b/anomalib/models/cflow/config.yaml
@@ -4,13 +4,14 @@ dataset:
   path: ./datasets/MVTec
   category: bottle
   task: segmentation
-  label_format: None
   image_size: 256
   train_batch_size: 16
   test_batch_size: 16
   inference_batch_size: 16
   fiber_batch_size: 64
-  num_workers: 36
+  num_workers: 8
+  transform_config: null
+  create_validation_set: false
 
 model:
   name: cflow

diff --git a/anomalib/models/dfkde/config.yaml b/anomalib/models/dfkde/config.yaml
@@ -4,11 +4,12 @@ dataset:
   path: ./datasets/MVTec
   category: bottle
   task: classification
-  label_format: None
   image_size: 256
   train_batch_size: 32
   test_batch_size: 32
   num_workers: 36
+  transform_config: null
+  create_validation_set: false
 
 model:
   name: dfkde

diff --git a/anomalib/models/dfm/config.yaml b/anomalib/models/dfm/config.yaml
@@ -4,11 +4,12 @@ dataset:
   path: ./datasets/MVTec
   category: bottle
   task: classification
-  label_format: None
   image_size: 256
   train_batch_size: 32
   test_batch_size: 32
   num_workers: 36
+  transform_config: null
+  create_validation_set: false
 
 model:
   name: dfm

diff --git a/anomalib/models/ganomaly/config.yaml b/anomalib/models/ganomaly/config.yaml
@@ -4,19 +4,20 @@ dataset:
   path: ./datasets/MVTec
   category: bottle
   task: classification
-  label_format: None
+  image_size: 256
+  train_batch_size: 32
+  test_batch_size: 32
+  inference_batch_size: 32
+  num_workers: 32
+  transform_config: null
+  create_validation_set: false
   tiling:
     apply: true
     tile_size: 64
     stride: null
     remove_border_count: 0
     use_random_tiling: False
     random_tile_count: 16
-  image_size: 256
-  train_batch_size: 32
-  test_batch_size: 32
-  inference_batch_size: 32
-  num_workers: 32
 
 model:
   name: ganomaly

diff --git a/anomalib/models/padim/config.yaml b/anomalib/models/padim/config.yaml
@@ -4,18 +4,19 @@ dataset:
   path: ./datasets/MVTec
   category: bottle
   task: segmentation
-  label_format: None
+  image_size: 256
+  train_batch_size: 32
+  test_batch_size: 32
+  num_workers: 36
+  transform_config: null
+  create_validation_set: false
   tiling:
     apply: false
     tile_size: null
     stride: null
     remove_border_count: 0
     use_random_tiling: False
     random_tile_count: 16
-  image_size: 256
-  train_batch_size: 32
-  test_batch_size: 32
-  num_workers: 36
 
 model:
   name: padim

diff --git a/anomalib/models/patchcore/config.yaml b/anomalib/models/patchcore/config.yaml
@@ -4,18 +4,19 @@ dataset:
   path: ./datasets/MVTec
   task: segmentation
   category: bottle
-  label_format: None
+  image_size: 224
+  train_batch_size: 32
+  test_batch_size: 1
+  num_workers: 36
+  transform_config: null
+  create_validation_set: false
   tiling:
-    apply: false
-    tile_size: null
+    apply: true
+    tile_size: 64
     stride: null
     remove_border_count: 0
     use_random_tiling: False
     random_tile_count: 16
-  image_size: 224
-  train_batch_size: 32
-  test_batch_size: 1
-  num_workers: 36
 
 model:
   name: patchcore

diff --git a/anomalib/models/stfpm/config.yaml b/anomalib/models/stfpm/config.yaml
@@ -4,19 +4,20 @@ dataset:
   path: ./datasets/MVTec
   category: bottle
   task: segmentation
-  label_format: None
+  image_size: 256
+  train_batch_size: 32
+  test_batch_size: 32
+  inference_batch_size: 32
+  num_workers: 36
+  transform_config: null
+  create_validation_set: false
   tiling:
     apply: false
     tile_size: null
     stride: null
     remove_border_count: 0
     use_random_tiling: False
     random_tile_count: 16
-  image_size: 256
-  train_batch_size: 32
-  test_batch_size: 32
-  inference_batch_size: 32
-  num_workers: 36
 
 model:
   name: stfpm