From 42eab9c3fa786d8a2092652909afc1bad7cd67de Mon Sep 17 00:00:00 2001 From: henrytsui000 Date: Fri, 31 May 2024 01:51:35 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=A9=B9=20[Update]=20logging,=20Temporaril?= =?UTF-8?q?y=20cancel=20graphviz?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- yolo/model/yolo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/yolo/model/yolo.py b/yolo/model/yolo.py index 0fce5c5..be65316 100644 --- a/yolo/model/yolo.py +++ b/yolo/model/yolo.py @@ -126,5 +126,5 @@ def get_model(cfg: Config) -> YOLO: model = YOLO(cfg.model, cfg.hyper.data.class_num) logger.info("✅ Success load model") log_model(model.model) - draw_model(model=model) + # draw_model(model=model) return model