Revert "Update yolo.py (#67)" (#68)

This reverts commit e7c0799.
micromind-toolkit · Dec 5, 2023 · 798c8dd · 798c8dd
1 parent e7c0799
commit 798c8dd
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/micromind/utils/yolo.py b/micromind/utils/yolo.py
@@ -10,6 +10,7 @@
 from collections import defaultdict
 from pathlib import Path
 
+import cv2
 import numpy as np
 import torch
 import torchvision
@@ -578,8 +579,6 @@ 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)