diff --git a/PytorchWildlife/models/detection/ultralytics_based/yolov5_base.py b/PytorchWildlife/models/detection/ultralytics_based/yolov5_base.py index 917b6d93c..3502ab013 100644 --- a/PytorchWildlife/models/detection/ultralytics_based/yolov5_base.py +++ b/PytorchWildlife/models/detection/ultralytics_based/yolov5_base.py @@ -14,7 +14,13 @@ from torch.utils.data import DataLoader from torch.hub import load_state_dict_from_url -from yolov5.utils.general import non_max_suppression, scale_boxes +try: + from yolov5.utils.general import non_max_suppression, scale_boxes +except ImportError: + # Older yolov5 releases expose this helper as ``scale_coords``; it was + # renamed to ``scale_boxes`` in newer versions (same signature). Import it + # under the expected name so both versions keep working (see issue #607). + from yolov5.utils.general import non_max_suppression, scale_coords as scale_boxes from ..base_detector import BaseDetector from ....data import transforms as pw_trans from ....data import datasets as pw_data