[YOLO] YOLOv8-custom 학습

1)Ultralytics 깃허브

GitHub - ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > OpenVINO > CoreML > TFLite

일단 Yolov5와 Yolov8 개발자는 같다!!

그렇기 때문에 사용감에 있어서 크게 다르지 않다.

 

일단 Roboflow에서 데이터셋이 준비되었다는 가정하에 진해함으로 이전 포스트를 차고 부탁드려요~

YOLO 학습용 Datasets 만들기 (Roboflow 활용법)

 

먼저 준비된 데이터셋에 있는 yaml파일에서 경로를 맞춰주고서 train 시작!!

 

#train 명령어
yolo detect train data=[dataset.yaml파일위치] model=yolov8n.pt epochs=100 imgsz=640 name=[폴더명] 

#predict 명령어
yolo detect predict model=[best.pt 파일경로] source=[적용해볼 동영상 파일경로]

*이외에도 train시 다양한 파라미터가 존재함으로 사용자가 확인하고 추가하여 좋은 모델을 만들면 된다. [여기]