Spaces:

henry000
/

YOLO

Running

henry000 commited on May 30, 2024

Commit

42eab9c

1 Parent(s): 6aabc6c

🩹 [Update] logging, Temporarily cancel graphviz

Files changed (1) hide show

yolo/model/yolo.py CHANGED Viewed

@@ -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)
     return model

     model = YOLO(cfg.model, cfg.hyper.data.class_num)
     logger.info("✅ Success load model")
     log_model(model.model)
+    # draw_model(model=model)
     return model