Add device log

app.py

@@ -5,6 +5,7 @@ from typing import List, Dict, Optional, Union
 import os
 import yaml
 
+from modules.logger_util import get_logger
 from modules.html_constants import (HEADER, DEFAULT_THEME, CSS)
 from modules.sam_inference import SamInference
 from modules.model_downloader import DEFAULT_MODEL_TYPE
@@ -15,6 +16,9 @@ from modules.constants import (AUTOMATIC_MODE, BOX_PROMPT_MODE, PIXELIZE_FILTER,
 from modules.video_utils import get_frames_from_dir
 
 
+logger = get_logger()
+
+
 class App:
     def __init__(self,
                  args: argparse.Namespace):
@@ -27,6 +31,7 @@ class App:
             model_dir=self.args.model_dir,
             output_dir=self.args.output_dir
         )
+        logger.info(f'device "{self.sam_inf.device}" is detected')
         self.image_modes = [AUTOMATIC_MODE, BOX_PROMPT_MODE]
         self.default_mode = BOX_PROMPT_MODE
         self.filter_modes = [PIXELIZE_FILTER, COLOR_FILTER]

modules/logger_util.py

@@ -8,5 +8,11 @@ def get_logger(name: Optional[str] = None, level: str = "INFO") -> logging.Logge
     logger = logging.getLogger(name)
     logger.setLevel(level.upper())
 
+    if not logger.handlers:
+        handler = logging.StreamHandler()
+        handler.setLevel(level.upper())
+
+        logger.addHandler(handler)
+
     return logger