supervision0.23 updated in requirements

PytorchWildlife/models/detection/ultralytics_based/__init__.py

@@ -1,4 +1,2 @@
 from .yolov5_base import *
-from .yolov8_base import *
-from .megadetectorv5 import *
-from .megadetectorv6 import *
+from .megadetectorv5 import *

PytorchWildlife/utils/post_process.py

@@ -36,7 +36,7 @@ def save_detection_images(results, output_dir, input_dir = None, overwrite=False
         overwrite (bool):
             Whether overwriting existing image folders. Default to False.
     """
-    box_annotator = sv.BoundingBoxAnnotator(thickness=4)
+    box_annotator = sv.BoxAnnotator(thickness=4)
     lab_annotator = sv.LabelAnnotator(text_color=sv.Color.BLACK, text_thickness=4, text_scale=2)
     os.makedirs(output_dir, exist_ok=True)
 

requirements.txt

@@ -3,7 +3,7 @@ torchvision
 torchaudio
 tqdm
 Pillow
-supervision==0.19.0
+supervision==0.23.0
 gradio
 ultralytics-yolov5
 chardet