Fix Logging (#719)
* Add logging setup * Fix fusing layers message * Fix logging does not have end * Add logging * Change logging to use logger * Update yolo.py I tried this in a cloned branch, and everything seems to work fine * Update yolo.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
1 parent
a925f283a7
commit
0892c44bc4
5 files changed
+13
-7
No files matched your search
@@ -13,7 +13,8 @@ from numpy import random
|
||||
from models.experimental import attempt_load
|
||||
from utils.datasets import LoadStreams, LoadImages
|
||||
from utils.general import (
|
||||
check_img_size, non_max_suppression, apply_classifier, scale_coords, xyxy2xywh, plot_one_box, strip_optimizer)
|
||||
check_img_size, non_max_suppression, apply_classifier, scale_coords,
|
||||
xyxy2xywh, plot_one_box, strip_optimizer, set_logging)
|
||||
from utils.torch_utils import select_device, load_classifier, time_synchronized
|
||||
|
||||
|
||||
@@ -23,6 +24,7 @@ def detect(save_img=False):
|
||||
webcam = source == '0' or source.startswith('rtsp') or source.startswith('http') or source.endswith('.txt')
|
||||
|
||||
# Initialize
|
||||
set_logging()
|
||||
device = select_device(opt.device)
|
||||
if os.path.exists(out):
|
||||
shutil.rmtree(out) # delete output folder
|
||||
|
||||
Reference in new issue
Block a user