Update processing/matting.py
Browse files- 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
|
|
|
|
| 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)
|