YOLOv8-seg 학습 1 : 테스트 데이터셋

조훈·2025년 6월 14일

1. Test Train

from ultralytics import YOLO

model = YOLO("yolov8n-seg.pt")

model.train(
    data="Test_train/data.yaml",
    epochs=100,
    imgsz=640,
    batch=16,
    workers=8,
    save=True,
    project="runs/segmentataion",
    name="a100_run",
    resume=False
)

  • 병원 데이터셋의 일부 ( train : 600장 / val : 70장 )을 테스트 학습

mAP가 약 0.46에 수렴하는 결과를 볼 수 있다. -> 개선 필요

0개의 댓글