EleFind-yolo11-elephant

from sahi import AutoDetectionModel
from sahi.predict import get_sliced_prediction
from huggingface_hub import hf_hub_download

model_path = hf_hub_download(
    repo_id="iamhelitha/EleFind-yolo11-elephant",
    filename="best.pt",
    repo_type="model",
)

model = AutoDetectionModel.from_pretrained(
    model_type="yolov8",  # SAHI uses 'yolov8' for YOLOv8/v11 models
    model_path=model_path,
    confidence_threshold=0.30,
    device="cpu",
)

result = get_sliced_prediction(
    image="aerial_image.jpg",
    detection_model=model,
    slice_height=1024,
    slice_width=1024,
    overlap_height_ratio=0.30,
    overlap_width_ratio=0.30,
    postprocess_type="NMS",
    postprocess_match_threshold=0.40,
)

print(f"Detected {len(result.object_prediction_list)} elephants")

from ultralytics import YOLO
from huggingface_hub import hf_hub_download

model_path = hf_hub_download(
    repo_id="iamhelitha/EleFind-yolo11-elephant",
    filename="best.pt",
    repo_type="model",
)

model = YOLO(model_path)
results = model.predict("aerial_image.jpg", conf=0.30)