✅ [Pass] tests, skip drawing if graphviz not found (#21)

Hi Ethan,

Thank you for your contribution! Your changes look great, and I appreciate the addition of the condition to skip drawing if Graphviz is not found. This will definitely help in environments where Graphviz isn't available.

I have reviewed the commit, and everything seems to be in order. All tests have passed as well, which is great. I'll go ahead and merge this PR.

Thanks again for your efforts!

Best regards,
Henry Tsui

Author: LaiEthanLai

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_structure(model.model)
-    # draw_model(model=model)
+    draw_model(model=model)
     return model

yolo/tools/drawer.py

@@ -95,6 +95,8 @@ def draw_model(*, model_cfg=None, model=None, v7_base=False):
         for jdx in range(idx, model_size):
             if model_mat[idx, jdx]:
                 dot.edge(str(idx), str(jdx))
-
-    dot.render("Model-arch", format="png", cleanup=True)
+    try:
+        dot.render("Model-arch", format="png", cleanup=True)
+    except:
+        logger.info("Warning: Could not find graphviz backend, continue without drawing the model architecture")
     logger.info("🎨 Drawing Model Architecture at Model-arch.png")