[23 하계 모각코] 2회차 목표 및 결과

vvon_joon·2023년 7월 15일
0

(23) Summer Study

목록 보기
3/7

목표

2회차: 23/07/15 13:00~16:00
계획: YOLOv8 모델 학습 (Segmentation)
방향: YOLO 중 가장 최신 모델인 YOLOv8을 이용해 Segmentation 수행

결과

Segmentation

학습

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n-seg.yaml')  # build a new model from YAML

# Train the model
model.train(data='coco128-seg.yaml', epochs=100, imgsz=640)
  • yolov8n 모델을 이용해 학습 진행
  • 학습을 위해 coco 데이터셋을 활용
  • train의 인자로 epochs, batch, imgsz 등을 설정할 수 있음

모델 구조

학습 결과

profile
김찬호 화이팅

0개의 댓글