mirror of
https://github.com/azaion/annotations.git
synced 2026-04-22 11:56:29 +00:00
Oleksandr Bezdieniezhnykh
58 lines
1.4 KiB
Cython
58 lines
1.4 KiB
Cython
import sys
|
|
|
|
from loguru import logger
|
|
|
|
cdef str CONFIG_FILE = "config.yaml" # Port for the zmq
|
|
|
|
cdef str QUEUE_CONFIG_FILENAME = "secured-config.json"
|
|
cdef str AI_ONNX_MODEL_FILE = "azaion.onnx"
|
|
|
|
cdef str CDN_CONFIG = "cdn.yaml"
|
|
cdef str MODELS_FOLDER = "models"
|
|
|
|
cdef int SMALL_SIZE_KB = 3
|
|
|
|
cdef str SPLIT_SUFFIX = "!split!"
|
|
cdef int TILE_DUPLICATE_CONFIDENCE_THRESHOLD = 5
|
|
|
|
logger.remove()
|
|
log_format = "[{time:HH:mm:ss} {level}] {message}"
|
|
logger.add(
|
|
sink="Logs/log_inference_{time:YYYYMMDD}.txt",
|
|
level="INFO",
|
|
format=log_format,
|
|
enqueue=True,
|
|
rotation="1 day",
|
|
retention="30 days",
|
|
)
|
|
logger.add(
|
|
sys.stdout,
|
|
level="DEBUG",
|
|
format=log_format,
|
|
filter=lambda record: record["level"].name in ("INFO", "DEBUG", "SUCCESS"),
|
|
colorize=True
|
|
)
|
|
logger.add(
|
|
sys.stderr,
|
|
level="WARNING",
|
|
format=log_format,
|
|
colorize=True
|
|
)
|
|
|
|
cdef log(str log_message):
|
|
logger.info(log_message)
|
|
|
|
cdef logerror(str error):
|
|
logger.error(error)
|
|
|
|
cdef format_time(int ms):
|
|
# Calculate hours, minutes, seconds, and hundreds of milliseconds.
|
|
h = ms // 3600000 # Total full hours.
|
|
ms_remaining = ms % 3600000
|
|
m = ms_remaining // 60000 # Full minutes.
|
|
ms_remaining %= 60000
|
|
s = ms_remaining // 1000 # Full seconds.
|
|
f = (ms_remaining % 1000) // 100 # Hundreds of milliseconds.
|
|
h = h % 10
|
|
return f"{h}{m:02}{s:02}{f}"
|