MogensR commited on
Commit
b1b89ac
·
1 Parent(s): d557b93

Update processing/matting.py

Browse files
Files changed (1) hide show
  1. processing/matting.py +3 -2
processing/matting.py CHANGED
@@ -12,7 +12,8 @@
12
  from dataclasses import dataclass
13
  import logging
14
 
15
- from utils.logger import setup_logger
 
16
  from utils.hardware.device_manager import DeviceManager
17
  from utils.config import ConfigManager
18
  from core.models import ModelFactory, ModelType
@@ -414,7 +415,7 @@ def composite(self, foreground: np.ndarray,
414
 
415
  Returns:
416
  Composited image
417
- """
418
  # Ensure alpha is 3-channel
419
  if len(alpha.shape) == 2:
420
  alpha = np.expand_dims(alpha, axis=2)
 
12
  from dataclasses import dataclass
13
  import logging
14
 
15
+ from utils.logger import get_logger
16
+ logger = get_logger(__name__)
17
  from utils.hardware.device_manager import DeviceManager
18
  from utils.config import ConfigManager
19
  from core.models import ModelFactory, ModelType
 
415
 
416
  Returns:
417
  Composited image
418
+ self.logger = get_logger(f"{__name__}.CompositingEngine")
419
  # Ensure alpha is 3-channel
420
  if len(alpha.shape) == 2:
421
  alpha = np.expand_dims(alpha, axis=2)