diff --git a/data/Objects365.yaml b/data/Objects365.yaml index 82b42a120d40..114bee2b159c 100644 --- a/data/Objects365.yaml +++ b/data/Objects365.yaml @@ -61,9 +61,9 @@ names: ['Person', 'Sneakers', 'Chair', 'Other Shoes', 'Hat', 'Car', 'Lamp', 'Gla # Download script/URL (optional) --------------------------------------------------------------------------------------- download: | from tqdm import tqdm - + from utils.general import Path, check_requirements, download, np, xyxy2xywhn - + check_requirements(('pycocotools>=2.0',)) from pycocotools.coco import COCO diff --git a/utils/general.py b/utils/general.py index 65dd9326797e..5316f504871a 100755 --- a/utils/general.py +++ b/utils/general.py @@ -82,11 +82,17 @@ def set_logging(name=None, verbose=VERBOSE): for h in logging.root.handlers: logging.root.removeHandler(h) # remove all handlers associated with the root logger object rank = int(os.getenv('RANK', -1)) # rank in world for Multi-GPU trainings - logging.basicConfig(format="%(message)s", level=logging.INFO if (verbose and rank in (-1, 0)) else logging.WARNING) - return logging.getLogger(name) + level = logging.INFO if (verbose and rank in (-1, 0)) else logging.WARNING + log = logging.getLogger(name) + log.setLevel(level) + handler = logging.StreamHandler() + handler.setFormatter(logging.Formatter("%(message)s")) + handler.setLevel(level) + log.addHandler(handler) -LOGGER = set_logging('yolov5') # define globally (used in train.py, val.py, detect.py, etc.) +set_logging() # run before defining LOGGER +LOGGER = logging.getLogger("yolov5") # define globally (used in train.py, val.py, detect.py, etc.) def user_config_dir(dir='Ultralytics', env_var='YOLOV5_CONFIG_DIR'):