JSON files are moved to a new directory ("jsonf_files")

json_config.py:
- Implemented `json_config_selector` function to handle JSON configurations for `augmentation`, `testing`, and `training` operations.
- Each operation now maps to specific configuration and schema JSON file paths via `JSON_FILES_PATHS.get_data_path`.
- Enables streamlined selection of JSON configurations, improving modularity.

augmentation.py, draw_rectangles.py, test.py, train.py
- Previous bindings for JSON_FILES_PATH replaced with json_config_selector()

data_paths.py:
- In ConfigFilePaths() paths are changed.

Author: richardRadli

config/data_paths.py

@@ -210,19 +210,19 @@ def get_data_path(self, key):
 class ConfigFilePaths(_Const):
     dirs_config_paths = {
         "config_augmentation":
-            "config/augmentation_config.json",
+            "config/json_files/augmentation_config.json",
         "config_schema_augmentation":
-            "config/augmentation_config_schema.json",
+            "config/json_files/augmentation_config_schema.json",
 
         "config_training":
-            "config/training_config.json",
+            "config/json_files/training_config.json",
         "config_schema_training":
-            "config/training_config_schema.json",
+            "config/json_files/training_config_schema.json",
 
         "config_testing":
-            "config/testing_config.json",
+            "config/json_files/testing_config.json",
         "config_schema_testing":
-            "config/testing_config_schema.json"
+            "config/json_files/testing_config_schema.json"
     }
 
     # ------------------------------------------------------------------------------------------------------------------

config/json_config.py

@@ -0,0 +1,20 @@
+from config.data_paths import JSON_FILES_PATHS
+
+
+def json_config_selector(operation):
+    json_cfg = {
+        "augmentation": {
+            "config": JSON_FILES_PATHS.get_data_path("config_augmentation"),
+            "schema": JSON_FILES_PATHS.get_data_path("config_schema_augmentation")
+        },
+        "testing": {
+            "config": JSON_FILES_PATHS.get_data_path("config_testing"),
+            "schema": JSON_FILES_PATHS.get_data_path("config_schema_testing")
+        },
+        "training": {
+            "config": JSON_FILES_PATHS.get_data_path("config_training"),
+            "schema": JSON_FILES_PATHS.get_data_path("config_schema_training")
+        }
+    }
+
+    return json_cfg[operation]

config/augmentation_config.json renamed to config/json_files/augmentation_config.json

@@ -7,8 +7,8 @@
 from typing import List, Tuple
 from tqdm import tqdm
 
-from config.data_paths import JSON_FILES_PATHS
 from config.dataset_config import dataset_images_path_selector
+from config.json_config import json_config_selector
 from utils.utils import load_config_json
 
 
@@ -174,8 +174,8 @@ def main() -> None:
     """
 
     aug_cfg = (
-        load_config_json(json_schema_filename=JSON_FILES_PATHS.get_data_path("config_schema_augmentation"),
-                         json_filename=JSON_FILES_PATHS.get_data_path("config_augmentation"))
+        load_config_json(json_schema_filename=json_config_selector("augmentation")["schema"],
+                         json_filename=json_config_selector("augmentation")["config"])
     )
 
     if aug_cfg.get("do_augmentation"):

config/augmentation_config_schema.json renamed to config/json_files/augmentation_config_schema.json

@@ -6,8 +6,8 @@
 from concurrent.futures import ProcessPoolExecutor
 from tqdm import tqdm
 
-from config.data_paths import JSON_FILES_PATHS
 from config.dataset_config import dataset_images_path_selector
+from config.json_config import json_config_selector
 from utils.utils import load_config_json, file_reader
 
 
@@ -56,8 +56,8 @@ def main() -> None:
 
     cfg = (
         load_config_json(
-            json_schema_filename=JSON_FILES_PATHS.get_data_path("config_schema_augmentation"),
-            json_filename=JSON_FILES_PATHS.get_data_path("config_augmentation")
+            json_schema_filename=json_config_selector("augmentation")["schema"],
+            json_filename=json_config_selector("augmentation")["config"]
         )
     )
 

config/testing_config.json renamed to config/json_files/testing_config.json

@@ -12,8 +12,8 @@
 from sklearn.metrics import mean_squared_error
 from tqdm import tqdm
 
-from config.data_paths import JSON_FILES_PATHS
 from config.network_config import network_configs
+from config.json_config import json_config_selector
 from config.dataset_config import dataset_data_path_selector, dataset_images_path_selector
 from models.network_selector import NetworkFactory
 from utils.utils import (setup_logger, device_selector, get_patch, patch2img, set_img_color, avg_of_list,
@@ -28,15 +28,15 @@ def __init__(self):
 
         train_cfg = (
             load_config_json(
-                json_schema_filename=JSON_FILES_PATHS.get_data_path("config_schema_training"),
-                json_filename=JSON_FILES_PATHS.get_data_path("config_training")
+                json_schema_filename=json_config_selector("training")["schema"],
+                json_filename=json_config_selector("training")["config"]
             )
         )
 
         self.test_cfg = (
             load_config_json(
-                json_schema_filename=JSON_FILES_PATHS.get_data_path("config_schema_testing"),
-                json_filename=JSON_FILES_PATHS.get_data_path("config_testing")
+                json_schema_filename=json_config_selector("testing")["schema"],
+                json_filename=json_config_selector("testing")["config"]
             )
         )
 

config/testing_config_schema.json renamed to config/json_files/testing_config_schema.json

@@ -13,7 +13,7 @@
 
 from config.network_config import network_configs
 from config.dataset_config import dataset_images_path_selector, dataset_data_path_selector
-from config.data_paths import JSON_FILES_PATHS
+from config.json_config import json_config_selector
 from dataloaders.data_loader_ae import MVTecDataset
 from dataloaders.data_loader_dae import MVTecDatasetDenoising
 from models.network_selector import NetworkFactory
@@ -31,8 +31,8 @@ def __init__(self):
 
         self.train_cfg = (
             load_config_json(
-                json_schema_filename=JSON_FILES_PATHS.get_data_path("config_schema_training"),
-                json_filename=JSON_FILES_PATHS.get_data_path("config_training")
+                json_schema_filename=json_config_selector("training")["schema"],
+                json_filename=json_config_selector("training")["config"]
             )
         )