23 lines
610 B
Python
23 lines
610 B
Python
import os
|
|
from cv_analysis.config import CONFIG
|
|
from cv_analysis.utils.display import save_mpl
|
|
|
|
LEVEL = CONFIG.visual_logging.level
|
|
OUTPUT_FOLDER = CONFIG.visual_logging.output_folder
|
|
|
|
|
|
class VisualLogger:
|
|
def __init__(self):
|
|
self.level_is_debug = LEVEL == "DEBUG"
|
|
self.output_folder = OUTPUT_FOLDER
|
|
if not os.path.exists(self.output_folder):
|
|
os.mkdir(self.output_folder)
|
|
|
|
def debug(self, img, name):
|
|
if self.level_is_debug:
|
|
output_path = os.path.join(self.output_folder, name)
|
|
save_mpl(img, output_path)
|
|
|
|
|
|
vizlogger = VisualLogger()
|