mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

Update hubconf.py to reset LOGGER.level after load (#8674)

Browse Source 
Resolves silent outputs after model load
This commit is contained in:
Glenn Jocher 2022-07-22 11:54:31 +02:00 committed by GitHub
parent 4c1784bd15
commit 38721de7ef
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
hubconf.py
View File

@ -34,6 +34,7 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
from utils.general import LOGGER, check_requirements, intersect_dicts, logging from utils.general import LOGGER, check_requirements, intersect_dicts, logging
from utils.torch_utils import select_device from utils.torch_utils import select_device
level = LOGGER.level
if not verbose: if not verbose:
LOGGER.setLevel(logging.WARNING) LOGGER.setLevel(logging.WARNING)
check_requirements(exclude=('tensorboard', 'thop', 'opencv-python')) check_requirements(exclude=('tensorboard', 'thop', 'opencv-python'))
@ -57,6 +58,7 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
model.names = ckpt['model'].names # set class names attribute model.names = ckpt['model'].names # set class names attribute
if autoshape: if autoshape:
model = AutoShape(model) # for file/URI/PIL/cv2/np inputs and NMS model = AutoShape(model) # for file/URI/PIL/cv2/np inputs and NMS
LOGGER.setLevel(level)
return model.to(device) return model.to(device)
except Exception as e: except Exception as e: