TIL 2023.05.22

YOLOv8 간단하게 써보기

 

출처: https://github.com/ultralytics/ultralytics

GitHub - ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > CoreML > TFLite

 

 

패키지 설치

pip install ultralytics

 

CLI(Command Line Interface)

yolo predict model=yolov8n.pt source='https://ultralytics.com/images/bus.jpg'

그냥 터미널에 입력하면 된다.

 

 

파이썬

from ultralytics import YOLO

# 모델 불러오기
model = YOLO("yolov8n.yaml")  # 처음부터 새로운 모델 생성
model = YOLO("yolov8n.pt")  # 사전 학습된 모델 불러오기 (학습에 권장됨)

# 모델 사용하기
model.train(data="coco128.yaml", epochs=3)  # 모델 학습
metrics = model.val()  # 검증 세트에서 모델 성능 평가
results = model("https://ultralytics.com/images/bus.jpg")  # 이미지에 대한 예측 수행
success = model.export(format="onnx")  # 모델을 ONNX 형식으로 내보내기

 

결과