YOLOv8 학습 중단 후 재개하는 법
YOLOv8을 이용하여 모델을 학습하다보면 학습 중단 후 다시 재개하여 학습햐여 하는 상황이 발생할 수 있습니다. 특히 colab을 사용하면 중간에 런타임이 끊기는 경우가 발생하기 때문에 다시 재개할 때 아래와 같은 과정을 거치면 다시 학습을 시작할 수 있습니다. 단계는 아래와 같습니다.1. 모델 로드중단된 시점의 모델 가중치 파일을 로드합니다. 일반적으로 학습 도중에는 체크포인트 파일이 저장되며, 이는 runs/ 디렉토리 내에 위치합니다.from ultralytics import YOLO# 기존에 학습된 모델 로드 (예: 'last.pt' 또는 'best.pt')model = YOLO('runs/detect/train/weights/last.pt') 2. 재학습 설정이전 학습에서 사용한 데이터 설정 파..
딥러닝
2024. 10. 31. 13:36
