Files
HSAP/algorithms/lane_ufld/code/CLRNet-main/clrnet/utils/logger.py

19 lines
542 B
Python
Raw Normal View History

import logging
def init_logger(log_file=None, log_level=logging.INFO):
stream_handler = logging.StreamHandler()
handlers = [stream_handler]
if log_file is not None:
file_handler = logging.FileHandler(log_file, 'w')
handlers.append(file_handler)
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(levelname)s - %(message)s')
for handler in handlers:
handler.setFormatter(formatter)
handler.setLevel(log_level)
logging.basicConfig(level=log_level, handlers=handlers)