ultralytics/yolov5
PyTorch 기반의 빠르고 정확한 객체 탐지 모델로, 산업 수준의 성능과 배포 간편성을 제공합니다.
57,630+56이번 주0 찜
17,482포크36이슈
중간맛 분석
PyTorch 기반 객체 탐지 프레임워크로, 사전 학습된 가중치(yolov5n/s/m/l/x)와 함께 제공됩니다. 웹캠, 이미지, 영상, YouTube, RTSP 스트림 등 다양한 입력을 지원하며, ONNX·CoreML·TensorRT·TFLite 등 여러 형식으로 내보낼 수 있습니다. 코드 몇 줄로 즉시 추론을 실행하거나, COCO 데이터셋으로 학습하는 예제도 제공되어 실무 배포가 용이합니다. 난이도는 중간 수준(대규모 GPU 학습 필요), 데모는 Google Colab·Kaggle에서 즉시 실행 가능합니다.
이런 레포예요
이럴 때 쓰면 좋아요
- 보안 카메라 영상에서 사람·차량 감지 및 실시간 경보 시스템 구축
- 모바일·엣지 기기(Jetson, iOS)에서 경량 객체 탐지 배포
- 자동차 자율주행 시스템의 도로 물체 인식 및 보행자 탐지
핵심 기능
다양한 크기의 사전 학습 모델(n/s/m/l/x) 제공웹캠·이미지·영상·스트림 등 다중 입력 소스 지원ONNX·CoreML·TensorRT·TFLite 등 다양한 형식으로의 모델 내보내기
대안 대비 차별점
YOLOv5는 복잡한 설정 없이 즉시 실행 가능한 프로덕션 수준의 정확도와 속도를 갖추고 있으며, PyTorch Hub 통합으로 별도의 설치 없이도 추론을 시작할 수 있다는 점에서 다른 YOLO 버전이나 객체탐지 프레임워크와 차별됩니다.
준비물
- Python>=3.8
- PyTorch>=1.8
바로 시작하기
- 리포지토리를 클론하고 의존성을 설치합니다. Python>=3.8과 PyTorch>=1.8이 필요합니다.
git clone https://github.com/ultralytics/yolov5
cd yolov5
pip install -r requirements.txt
- PyTorch Hub를 통해 사전 학습 모델을 로드하고 추론을 실행합니다.
import torch
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
img = 'https://ultralytics.com/images/zidane.jpg'
results = model(img)
results.print()
- 또는 detect.py 스크립트로 웹캠에서 실시간 추론을 합니다.
python detect.py --weights yolov5s.pt --source 0
- 로컬 이미지 파일에 대해 추론합니다.
python detect.py --weights yolov5s.pt --source img.jpg
- 자신의 데이터로 모델을 학습시킵니다. COCO 데이터셋에서 300 에포크로 YOLOv5s를 학습합니다.
python train.py --data coco.yaml --epochs 300 --weights '' --cfg yolov5s.yaml --batch-size 64
- 학습된 모델을 ONNX·CoreML·TensorRT 등으로 내보냅니다(상세는 공식 문서 참조).
python export.py --weights yolov5s.pt --include onnx
별 추이 · 7일
57,630+56 / 7일
활용성 · 따라 만들기 좋은가
따라 하기 좋음샘플 점수 44/100
- 관대한 라이선스
- 테스트 있음
- 예제 디렉토리
- 최근 활동
- AGENTS.md
- llms.txt
- 템플릿
🟢 최근 활동 있음 · 2020년 생성
이 레포에 깃밥 배지 달기
[](https://www.gitbap.com/r/ultralytics/yolov5)🔌 클로드에서 바로 써보기
깃밥을 클로드에 연결하면 이런 레포를 클로드·커서·Codex에서 바로 찾고 체험할 수 있어요.
방법 ① 커넥터 URL
클로드 설정 → 커넥터에 이 주소를 붙여넣기.
https://www.gitbap.com/api/mcp방법 ② 플러그인 (체험까지)
플러그인을 설치하면 /gitbap-trending·/gitbap-try 슬래시 커맨드로 레포를 그 자리에서 체험. Claude Code에서 아래 두 줄을 차례로 실행하세요.
/plugin marketplace add jakeparkcolde/gitbap-cowork/plugin install gitbap-cowork@gitbapAI가 README 기반으로 요약했습니다 · 원문 보기