1. 모델 선택

flutter는 YOLOv5, YOLOv8지원

이전 버전들에 비해 경량화, 성능이 좋은 YOLOv8 선택

2. 모델 학습

YOLOv8n부터 학습

• 테스트용으로 local에서 학습 시작

from ultralytics import YOLO
import os

os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE' # 라이브러리 충돌 방지

model = YOLO('yolov8n.pt')

# 모델 학습

model.train(data='...\\\\street-facilities\\\\street-facilities.yaml', epochs=50, imgsz=640)

3. 학습 결과

데이터 구성

• 참고 - COCO 데이터셋 yolov8 벤치마크 지표