프로젝트 소개
이 저장소는 Ultralytics YOLO 모델을 다양한 컴퓨터 비전 작업에 활용할 수 있도록 PyQt로 구축된 그래픽 사용자 인터페이스(GUI) 애플리케이션을 제공합니다. 객체 감지, 인간 포즈 추정, 분할, 방향성 경계 상자(ORiented Bounding Boxes)를 지원합니다. 이 애플리케이션은 로컬 이미지/비디오 파일, 카메라 또는 RTSP 스트림에서 입력을 처리할 수 있습니다.
주요 기능은 다음과 같습니다:
- **AI 작업**: 감지, 포즈 추정, 분할, 방향성 경계 상자.
- **지원 모델**: YOLO26 (n, s, m, l, x), YOLOv13 (n, s, l, x), YOLO11 (n, s, m, l, x). 모든 스크립트는 ONNX로 내보낸 YOLO 모델을 사용합니다.
- **트래커**: 객체 추적을 위한 DeepSort 및 ByteTrack.
- **입력 소스**: 로컬 파일(이미지/비디오), 카메라, RTSP 스트림.
**설치**:
- pip 사용: `pip install -r requirements.txt`
- conda 사용: `conda env create -f environment.yml` 후 `conda activate yolo_gui`로 환경 활성화.
**가중치 다운로드**: `python download_weights.py`를 실행하여 모델 가중치를 가져오며, `weights/` 폴더에 저장됩니다.
**실행**: `python main.py`를 실행하여 GUI를 시작합니다.
저장소에는 `data/ui.png`에 GUI 스크린샷이 포함되어 있습니다. 코드를 직접 작성하지 않고 YOLO 기반 감지 및 추적을 적용할 수 있는 사용자 친화적인 인터페이스가 필요한 사용자를 위해 설계되었습니다.
Comments
0 people shared their preference · Deer Point appears after 10 participants
Sign in to join the discussion.