e7b28f5bda
Merge in RR/cv-analysis from remove_pil to master
Squashed commit of the following:
commit 83c8d88f3d48404251470176c70979ee75ae068b
Author: Julius Unverfehrt <julius.unverfehrt@iqser.com>
Date: Thu Jul 21 10:51:51 2022 +0200
remove deprecated server tests
commit cebc03b5399ac257a74036b41997201f882f5b74
Author: Julius Unverfehrt <julius.unverfehrt@iqser.com>
Date: Thu Jul 21 10:51:08 2022 +0200
remove deprecated server tests
commit ce2845b0c51f001b7b5b8b195d6bf7e034ec4e39
Author: Julius Unverfehrt <julius.unverfehrt@iqser.com>
Date: Wed Jul 20 17:05:00 2022 +0200
repair tests to work without pillow WIP
commit 023fdab8322f28359a24c63e32635a3d0deccbe4
Author: Isaac Riley <Isaac.Riley@iqser.com>
Date: Wed Jul 20 16:40:36 2022 +0200
fixed typo
commit 33850ca83a175f74789ae6b9bebd057ed84b7fb3
Author: Isaac Riley <Isaac.Riley@iqser.com>
Date: Wed Jul 20 16:38:37 2022 +0200
fixed import from refactored open_img.py
commit dbc6d345f074e538948e2c4f94ebed8a5ef520bc
Author: Isaac Riley <Isaac.Riley@iqser.com>
Date: Wed Jul 20 16:32:42 2022 +0200
removed PIL from production code, now inly in scripts
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import os
|
|
from cv_analysis.config import CONFIG
|
|
from cv_analysis.utils.display import save_image
|
|
|
|
|
|
class VisualLogger:
|
|
def __init__(self, level, output_folder):
|
|
self.level = level
|
|
self.output_folder = output_folder
|
|
if not os.path.exists(self.output_folder):
|
|
os.mkdir(self.output_folder)
|
|
|
|
def _save(self, img, name):
|
|
output_path = os.path.join(self.output_folder, name)
|
|
save_image(img, output_path)
|
|
|
|
def info(self, img, name):
|
|
if self._level_is_info():
|
|
self._save(img, name)
|
|
|
|
def debug(self, img, name):
|
|
if self._level_is_debug():
|
|
self._save(img, name)
|
|
|
|
def all(self, img, name):
|
|
if self._level_is_debug():
|
|
self._save(img, name)
|
|
|
|
def _level_is_info(self):
|
|
return self.level in {"INFO", "DEBUG", "ALL"}
|
|
|
|
def _level_is_debug(self):
|
|
return self.level in {"DEBUG", "ALL"}
|
|
|
|
def _level_is_all(self):
|
|
return self.level == "ALL"
|
|
|
|
|
|
vizlogger = VisualLogger(CONFIG.visual_logging.level, CONFIG.visual_logging.output_folder)
|