Update yolo.py (#67)

* Update yolo.py * fix linters
micromind-toolkit · Dec 5, 2023 · e7c0799 · e7c0799
1 parent 9a27a37
commit e7c0799
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/micromind/utils/yolo.py b/micromind/utils/yolo.py
@@ -10,7 +10,6 @@
 from collections import defaultdict
 from pathlib import Path
 
-import cv2
 import numpy as np
 import torch
 import torchvision
@@ -579,6 +578,8 @@ def draw_bounding_boxes_and_save(
     -------
         None
     """
+    import cv2  # here to avoid extra dependecy
+
     color_dict = {
         label: tuple(
             (((i + 1) * 50) % 256, ((i + 1) * 100) % 256, ((i + 1) * 150) % 256)