improved `InferenceSlicer` docstring

This commit is contained in:
SkalskiP 2025-11-15 00:00:52 +01:00
parent 12d7c369db
commit a296712000
1 changed files with 4 additions and 0 deletions

View File

@ -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)