SeieunYoo/yolov8n-custom-1

Yolov8-30000-epoch3

Loss

# 이미지 3만장 학습
import torch
from ultralytics import YOLO

# 사전 학습된 YOLOv5 모델 로드
model = YOLO('yolov8n.pt')

# 학습 설정
epochs = 3
data_path = 'data/data.yaml'
img_size = 640
batch_size = 10
project_name = 'yolov8n_custom'
project_dir = 'runs/train'

# 사용자 정의 학습 함수
def train_yolov8(model, data_path, img_size, batch_size, epochs, project_name, project_dir):
    results = model.train(
        data=data_path,
        imgsz=img_size,
        epochs=epochs,
        batch=batch_size,
        name=project_name,
        project=project_dir,
        warmup_epochs=1.0,
        box=0.02,
        mosaic=0.5,  
    )

# 모델 학습
train_yolov8(model, data_path, img_size, batch_size, epochs, project_name, project_dir)