Ultralytics Docs: Using YOLO11 with SAHI for Sliced Inference

Welcome to the Ultralytics documentation on how to use YOLO11 with SAHI (Slicing Aided Hyper Inference). This comprehensive guide aims to furnish you with all the essential knowledge you'll need to implement SAHI alongside YOLO11. We'll deep-dive into what SAHI is, why sliced inference is critical for large-scale applications, and how to integrate these functionalities with YOLO11 for enhanced object detection performance.

SAHI 소개

SAHI(슬라이싱 지원 하이퍼 추론)는 대규모 고해상도 이미지의 객체 감지 알고리즘을 최적화하도록 설계된 혁신적인 라이브러리입니다. 핵심 기능은 이미지를 관리 가능한 슬라이스로 분할하고 각 슬라이스에서 객체 감지를 실행한 다음 결과를 다시 연결하는 것입니다. SAHI는 YOLO 시리즈를 비롯한 다양한 물체 감지 모델과 호환되므로 컴퓨팅 리소스를 최적화하여 사용할 수 있는 유연성을 제공합니다.

Watch: Inference with SAHI (Slicing Aided Hyper Inference) using Ultralytics YOLO11

SAHI의 주요 기능

• 원활한 통합: SAHI는 YOLO 모델과 손쉽게 통합되므로 많은 코드 수정 없이도 슬라이싱 및 탐지를 시작할 수 있습니다.
• 리소스 효율성: SAHI는 큰 이미지를 작은 부분으로 나누어 메모리 사용량을 최적화하여 리소스가 제한된 하드웨어에서 고품질 탐지를 실행할 수 있도록 합니다.
• High Accuracy: SAHI maintains the detection accuracy by employing smart algorithms to merge overlapping detection boxes during the stitching process.

슬라이스 추론이란 무엇인가요?

슬라이스 추론은 큰 이미지나 고해상도 이미지를 작은 세그먼트(슬라이스)로 세분화하고, 이러한 슬라이스에서 객체 감지를 수행한 다음, 슬라이스를 다시 컴파일하여 원본 이미지에서 객체 위치를 재구성하는 방법을 말합니다. 이 기술은 컴퓨팅 리소스가 제한되어 있거나 메모리 문제가 발생할 수 있는 초고해상도 이미지로 작업할 때 매우 유용합니다.

슬라이스 추론의 이점

• 컴퓨팅 부담 감소: 이미지 슬라이스가 작을수록 처리 속도가 빨라지고 메모리 사용량이 적어져 저사양 하드웨어에서 더 원활하게 작동합니다.

• 보존된 감지 품질: 각 슬라이스는 독립적으로 처리되므로, 슬라이스가 관심 있는 물체를 캡처할 수 있을 만큼 충분히 크다면 물체 감지 품질이 저하되지 않습니다.

• 향상된 확장성: 이 기술을 사용하면 이미지의 크기와 해상도에 따라 물체 감지 기능을 보다 쉽게 확장할 수 있으므로 위성 이미지에서 의료 진단에 이르기까지 다양한 애플리케이션에 이상적입니다.

YOLO11 without SAHI	YOLO11 with SAHI

설치 및 준비

설치

시작하려면 최신 버전의 SAHI 및 Ultralytics 을 설치하세요:

pip install -U ultralytics sahi

모듈 가져오기 및 리소스 다운로드

Here's how to import the necessary modules and download a YOLO11 model and some test images:

from sahi.utils.file import download_from_url
from sahi.utils.yolov8 import download_yolov8s_model

# Download YOLO11 model
model_path = "models/yolo11s.pt"
download_yolov8s_model(model_path)

# Download test images
download_from_url(
    "https://raw.githubusercontent.com/obss/sahi/main/demo/demo_data/small-vehicles1.jpeg",
    "demo_data/small-vehicles1.jpeg",
)
download_from_url(
    "https://raw.githubusercontent.com/obss/sahi/main/demo/demo_data/terrain2.png",
    "demo_data/terrain2.png",
)

Standard Inference with YOLO11

모델 인스턴스화

You can instantiate a YOLO11 model for object detection like this:

from sahi import AutoDetectionModel

detection_model = AutoDetectionModel.from_pretrained(
    model_type="yolov8",
    model_path=yolov8_model_path,
    confidence_threshold=0.3,
    device="cpu",  # or 'cuda:0'
)

표준 예측 수행

이미지 경로 또는 덩어리 이미지를 사용하여 표준 추론을 수행합니다.

from sahi.predict import get_prediction

# With an image path
result = get_prediction("demo_data/small-vehicles1.jpeg", detection_model)

# With a numpy image
result = get_prediction(read_image("demo_data/small-vehicles1.jpeg"), detection_model)

결과 시각화

예측된 바운딩 박스와 마스크를 내보내고 시각화합니다:

result.export_visuals(export_dir="demo_data/")
Image("demo_data/prediction_visual.png")

Sliced Inference with YOLO11

슬라이스 크기와 겹침 비율을 지정하여 슬라이스 추론을 수행합니다:

from sahi.predict import get_sliced_prediction

result = get_sliced_prediction(
    "demo_data/small-vehicles1.jpeg",
    detection_model,
    slice_height=256,
    slice_width=256,
    overlap_height_ratio=0.2,
    overlap_width_ratio=0.2,
)

예측 결과 처리

SAHI는 PredictionResult 객체를 다양한 어노테이션 형식으로 변환할 수 있습니다:

# Access the object prediction list
object_prediction_list = result.object_prediction_list

# Convert to COCO annotation, COCO prediction, imantics, and fiftyone formats
result.to_coco_annotations()[:3]
result.to_coco_predictions(image_id=1)[:3]
result.to_imantics_annotations()[:3]
result.to_fiftyone_detections()[:3]

배치 예측

이미지 디렉토리에 대한 일괄 예측의 경우:

from sahi.predict import predict

predict(
    model_type="yolov8",
    model_path="path/to/yolo11n.pt",
    model_device="cpu",  # or 'cuda:0'
    model_confidence_threshold=0.4,
    source="path/to/dir",
    slice_height=256,
    slice_width=256,
    overlap_height_ratio=0.2,
    overlap_width_ratio=0.2,
)

That's it! Now you're equipped to use YOLO11 with SAHI for both standard and sliced inference.

인용 및 감사

연구 또는 개발 작업에 SAHI를 사용하는 경우, 원본 SAHI 논문을 인용하고 저자를 인정해 주세요:

@article{akyon2022sahi,
  title={Slicing Aided Hyper Inference and Fine-tuning for Small Object Detection},
  author={Akyon, Fatih Cagatay and Altinuc, Sinan Onur and Temizel, Alptekin},
  journal={2022 IEEE International Conference on Image Processing (ICIP)},
  doi={10.1109/ICIP46576.2022.9897990},
  pages={966-970},
  year={2022}
}

We extend our thanks to the SAHI research group for creating and maintaining this invaluable resource for the computer vision community. For more information about SAHI and its creators, visit the SAHI GitHub repository.

자주 묻는 질문

How can I integrate YOLO11 with SAHI for sliced inference in object detection?

Integrating Ultralytics YOLO11 with SAHI (Slicing Aided Hyper Inference) for sliced inference optimizes your object detection tasks on high-resolution images by partitioning them into manageable slices. This approach improves memory usage and ensures high detection accuracy. To get started, you need to install the ultralytics and sahi libraries:

pip install -U ultralytics sahi

Then, download a YOLO11 model and test images:

from sahi.utils.file import download_from_url
from sahi.utils.yolov8 import download_yolov8s_model

# Download YOLO11 model
model_path = "models/yolo11s.pt"
download_yolov8s_model(model_path)

# Download test images
download_from_url(
    "https://raw.githubusercontent.com/obss/sahi/main/demo/demo_data/small-vehicles1.jpeg",
    "demo_data/small-vehicles1.jpeg",
)

자세한 지침은 슬라이스 추론 가이드를 참조하세요.

Why should I use SAHI with YOLO11 for object detection on large images?

Using SAHI with Ultralytics YOLO11 for object detection on large images offers several benefits:

• 컴퓨팅 부담 감소: 슬라이스가 작을수록 처리 속도가 빠르고 메모리 사용량이 적어 리소스가 제한된 하드웨어에서도 고품질 탐지를 실행할 수 있습니다.
• 탐지 정확도 유지: SAHI는 지능형 알고리즘을 사용하여 겹치는 상자를 병합하여 탐지 품질을 유지합니다.
• 향상된 확장성: 다양한 이미지 크기와 해상도에 걸쳐 물체 감지 작업을 확장할 수 있어 위성 이미지 분석 및 의료 진단과 같은 다양한 애플리케이션에 이상적입니다.

문서에서 슬라이스 추론의 이점에 대해 자세히 알아보세요.

Can I visualize prediction results when using YOLO11 with SAHI?

Yes, you can visualize prediction results when using YOLO11 with SAHI. Here's how you can export and visualize the results:

from IPython.display import Image

result.export_visuals(export_dir="demo_data/")
Image("demo_data/prediction_visual.png")

이 명령은 시각화된 예측을 지정된 디렉터리에 저장한 다음 이미지를 로드해 노트북이나 애플리케이션에서 볼 수 있습니다. 자세한 가이드는 표준 추론 섹션을 참조하세요.

What features does SAHI offer for improving YOLO11 object detection?

SAHI (Slicing Aided Hyper Inference) offers several features that complement Ultralytics YOLO11 for object detection:

• 원활한 통합: SAHI는 YOLO 모델과 쉽게 통합되므로 최소한의 코드 조정만 필요합니다.
• 리소스 효율성: 큰 이미지를 작은 조각으로 분할하여 메모리 사용량과 속도를 최적화합니다.
• 높은 정확도: 스티칭 과정에서 겹치는 감지 상자를 효과적으로 병합함으로써 SAHI는 높은 감지 정확도를 유지합니다.

더 자세히 알아보려면 SAHI의 주요 기능에 대해 읽어보세요.

How do I handle large-scale inference projects using YOLO11 and SAHI?

To handle large-scale inference projects using YOLO11 and SAHI, follow these best practices:

1. 필수 라이브러리를 설치합니다: ultralytics 및 sahi의 최신 버전이 설치되어 있는지 확인합니다.
2. 슬라이스 추론을 구성합니다: 특정 프로젝트에 맞는 최적의 슬라이스 크기와 겹침 비율을 결정합니다.
3. 일괄 예측 실행: SAHI의 기능을 사용하여 이미지 디렉토리에 대한 일괄 예측을 수행하면 효율성이 향상됩니다.

배치 예측의 예입니다:

from sahi.predict import predict

predict(
    model_type="yolov8",
    model_path="path/to/yolo11n.pt",
    model_device="cpu",  # or 'cuda:0'
    model_confidence_threshold=0.4,
    source="path/to/dir",
    slice_height=256,
    slice_width=256,
    overlap_height_ratio=0.2,
    overlap_width_ratio=0.2,
)

자세한 단계는 배치 예측 섹션을 참조하세요.

📅 Created 11 months ago ✏️ Updated 12 days ago

댓글