From 46928b4dc99c5d70b3c35d7046f723752e33dc22 Mon Sep 17 00:00:00 2001 From: Daniela Ruiz Date: Mon, 9 Sep 2024 17:11:25 +0000 Subject: [PATCH] supervision0.23 updated in requirements --- .../models/detection/ultralytics_based/__init__.py | 4 +--- PytorchWildlife/utils/post_process.py | 2 +- requirements.txt | 2 +- 3 files changed, 3 insertions(+), 5 deletions(-) diff --git a/PytorchWildlife/models/detection/ultralytics_based/__init__.py b/PytorchWildlife/models/detection/ultralytics_based/__init__.py index 0e07b779..b81b3b35 100644 --- a/PytorchWildlife/models/detection/ultralytics_based/__init__.py +++ b/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 * \ No newline at end of file +from .megadetectorv5 import * \ No newline at end of file diff --git a/PytorchWildlife/utils/post_process.py b/PytorchWildlife/utils/post_process.py index 0f8a7a7a..11d73583 100644 --- a/PytorchWildlife/utils/post_process.py +++ b/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) diff --git a/requirements.txt b/requirements.txt index 997f2b18..3855f4ea 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ torchvision torchaudio tqdm Pillow -supervision==0.19.0 +supervision==0.23.0 gradio ultralytics-yolov5 chardet \ No newline at end of file