🚚 [Rename] tools and utils, move the function!

model/yolo.py

@@ -1,26 +1,10 @@
-import inspect
 from typing import Any, Dict, List, Union
 
 import torch
 import torch.nn as nn
 from loguru import logger
 from omegaconf import OmegaConf
-
-from model import module
-from utils.tools import load_model_cfg
-
-
-def get_layer_map():
-    """
-    Dynamically generates a dictionary mapping class names to classes,
-    filtering to include only those that are subclasses of nn.Module,
-    ensuring they are relevant neural network layers.
-    """
-    layer_map = {}
-    for name, obj in inspect.getmembers(module, inspect.isclass):
-        if issubclass(obj, nn.Module) and obj is not nn.Module:
-            layer_map[name] = obj
-    return layer_map
+from tools.layer_helper import get_layer_map
 
 
 class YOLO(nn.Module):

tools/layer_helper.py

@@ -0,0 +1,20 @@
+import inspect
+import torch.nn as nn
+from model import module
+
+
+def auto_pad():
+    raise NotImplementedError
+
+
+def get_layer_map():
+    """
+    Dynamically generates a dictionary mapping class names to classes,
+    filtering to include only those that are subclasses of nn.Module,
+    ensuring they are relevant neural network layers.
+    """
+    layer_map = {}
+    for name, obj in inspect.getmembers(module, inspect.isclass):
+        if issubclass(obj, nn.Module) and obj is not nn.Module:
+            layer_map[name] = obj
+    return layer_map

tools/log_helper.py

@@ -0,0 +1,23 @@
+"""
+Module for initializing logging tools used in machine learning and data processing. 
+Supports integration with Weights & Biases (wandb), Loguru, TensorBoard, and other 
+logging frameworks as needed.
+
+This setup ensures consistent logging across various platforms, facilitating 
+effective monitoring and debugging.
+
+Example:
+    from tools.logger import custom_logger
+    custom_logger()
+"""
+
+import sys
+from loguru import logger
+
+
+def custom_logger():
+    logger.remove()
+    logger.add(
+        sys.stderr,
+        format="<green>{time:MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <level>{message}</level>",
+    )

train.py

@@ -1,7 +1,6 @@
-import argparse
 from loguru import logger
 from model.yolo import get_model
-from utils.tools import load_model_cfg, custom_logger
+from tools.log_helper import custom_logger
 from utils.get_dataset import download_coco_dataset
 import hydra
 from config.config import Config

utils/get_dataset.py

@@ -78,7 +78,7 @@ def download_coco_dataset(download_cfg):
 
 
 if __name__ == "__main__":
-    from tools import custom_logger
+    from tools.log_helper import custom_logger
 
     custom_logger()
     download_coco_dataset()

utils/tools.py

@@ -1,72 +0,0 @@
-import os
-import sys
-import yaml
-from loguru import logger
-from typing import Dict, Any
-
-
-def complete_path(file_name: str = "v7-base.yaml") -> str:
-    """
-    Ensures the path to a model configuration is a existing file
-
-    Parameters:
-        file_name (str): The filename or path, with default 'v7-base.yaml'.
-
-    Returns:
-        str: A complete path with necessary prefix and extension.
-    """
-    # Ensure the file has the '.yaml' extension if missing
-    if not file_name.endswith(".yaml"):
-        file_name += ".yaml"
-
-    # Add folder prefix if only the filename is provided
-    if os.path.dirname(file_name) == "":
-        file_name = os.path.join("./config/model", file_name)
-
-    return file_name
-
-
-def load_model_cfg(file_path: str) -> Dict[str, Any]:
-    """
-    Read a YAML configuration file, ensure necessary keys are present, and return its content as a dictionary.
-
-    Args:
-        file_path (str): The path to the YAML configuration file.
-
-    Returns:
-        Dict[str, Any]: The contents of the YAML file as a dictionary.
-
-    Raises:
-        FileNotFoundError: If the YAML file cannot be found.
-        yaml.YAMLError: If there is an error parsing the YAML file.
-    """
-    file_path = complete_path(file_path)
-    try:
-        with open(file_path, "r") as file:
-            model_cfg = yaml.safe_load(file) or {}
-
-        # Check for required keys and set defaults if not present
-        if "nc" not in model_cfg:
-            model_cfg["nc"] = 80
-            logger.warning("'nc' not found in the YAML file. Setting default 'nc' to 80.")
-
-        if "model" not in model_cfg:
-            logger.error("'model' is missing in the configuration file.")
-            raise ValueError("Missing required key: 'model'")
-
-        return model_cfg
-
-    except FileNotFoundError:
-        logger.error(f"YAML file not found: {file_path}")
-        raise
-    except yaml.YAMLError as e:
-        logger.error(f"Error parsing YAML file: {e}")
-        raise
-
-
-def custom_logger():
-    logger.remove()
-    logger.add(
-        sys.stderr,
-        format="<green>{time:MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <level>{message}</level>",
-    )