diff --git a/supervision/detection/tools/inference_slicer.py b/supervision/detection/tools/inference_slicer.py index 103b3fa4..4cc05f19 100644 --- a/supervision/detection/tools/inference_slicer.py +++ b/supervision/detection/tools/inference_slicer.py @@ -83,6 +83,8 @@ class InferenceSlicer: import supervision as sv from rfdetr import RFDETRMedium + model = RFDETRMedium() + def callback(tile): return model.predict(tile) @@ -97,6 +99,8 @@ class InferenceSlicer: from PIL import Image from ultralytics import YOLO + model = YOLO("yolo11m.pt") + def callback(tile): results = model(tile)[0] return sv.Detections.from_ultralytics(results)