NVIDIA DeepStream 9.0에서 Ultralytics YOLO OBB 실행하기

섹션별 상세

사용자는 NVIDIA DeepStream 9.0에서 YOLOv8-OBB 및 YOLO11-OBB 모델을 구동하는 파이프라인을 구축했다. 이 구현은 별도의 C++나 Python 커스텀 코드 작성 없이 설정 파일만으로 작동한다.

회전된 바운딩 박스 시각화는 DeepStream의 기본 플러그인인 nvdsosd를 활용하여 처리한다. 이를 통해 별도의 렌더링 로직 구현 없이 네이티브 환경에서 OBB 결과를 그릴 수 있다.

해당 프로젝트는 DOTA 항공 이미지 데이터셋을 사용하여 선박, 비행기, 차량 등 다양한 객체 탐지 성능을 검증했다. 파일 입력 및 RTSP 스트림 환경을 모두 지원한다.

회전된 바운딩 박스(OBB): — Oriented Bounding Box의 약자로, 일반적인 수평 바운딩 박스와 달리 객체의 회전 각도를 포함하여 탐지하는 방식이다. 항공 이미지나 기울어진 객체가 많은 환경에서 정확한 위치 파악을 위해 사용된다.
딥스트림(DeepStream): — NVIDIA에서 제공하는 스트리밍 분석 SDK로, AI 기반의 영상 분석 파이프라인을 구축하는 데 최적화되어 있다. 하드웨어 가속을 통해 실시간 추론 및 처리를 지원한다.
DOTA 데이터셋(DOTA): — Dataset for Object Detection in Aerial images의 약자로, 항공 이미지에서 객체를 탐지하기 위한 대규모 데이터셋이다. 다양한 각도와 크기의 객체가 포함되어 있어 OBB 모델 학습 및 평가에 주로 활용된다.

DeepStream추천

AI 추론 및 스트리밍 분석

YOLOv8추천

객체 탐지 모델

YOLO11추천

객체 탐지 모델