improved `InferenceSlicer` docstring
This commit is contained in:
parent
12d7c369db
commit
a296712000
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Reference in New Issue