🧑‍💻 [add] custom logger, new format of logging

train.py

@@ -1,8 +1,7 @@
 import argparse
-import sys
 from loguru import logger
 from model.yolo import get_model
-from utils.tools import load_model_cfg
+from utils.tools import load_model_cfg, custom_logger
 
 
 def parse_arguments() -> argparse.Namespace:
@@ -14,12 +13,13 @@ def parse_arguments() -> argparse.Namespace:
     """
     parser = argparse.ArgumentParser(description="Load a YOLO model configuration and display the model.")
     parser.add_argument(
-        "--model-config", type=str, default="v7-base.yaml", help="Name or path to the model configuration file."
+        "--model-config", type=str, default="v7-base", help="Name or path to the model configuration file."
     )
     return parser.parse_args()
 
 
 if __name__ == "__main__":
+    custom_logger()
     args = parse_arguments()
     model_cfg = load_model_cfg(args.model_config)
     model = get_model(model_cfg)

utils/tools.py

@@ -1,4 +1,5 @@
 import os
+import sys
 import yaml
 from loguru import logger
 from typing import Dict, Any
@@ -49,10 +50,6 @@ def load_model_cfg(file_path: str) -> Dict[str, Any]:
             model_cfg["nc"] = 80
             logger.warning("'nc' not found in the YAML file. Setting default 'nc' to 80.")
 
-        if "anchor" not in model_cfg:
-            logger.error("'anchor' is missing in the configuration file.")
-            raise ValueError("Missing required key: 'anchor'")
-
         if "model" not in model_cfg:
             logger.error("'model' is missing in the configuration file.")
             raise ValueError("Missing required key: 'model'")
@@ -65,3 +62,11 @@ def load_model_cfg(file_path: str) -> Dict[str, Any]:
     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>",
+    )