YOLO (2)

Chapter 2 Classification

• 데이터 셋 확보
  Rock Paper Scissors Dataset

• 구글 코랩 환경에서 데이터 셋 저장할 폴더 만들고 다운로드

!mkdir dataset

!wget Raw URL입력 -O ./dataset/dataset.zip

• 압축 풀기

!unzip ./dataset/dataset.zip -d ./dataset

• 별도 학습 없이 예측시켜보기

from ultralytics import YOLO

model = YOLO('yolov8n-cls.pt')
model.predict(source='./rock_sample.jpg', save=True)

• 학습

model = YOLO('yolov8n-cls.pt')
results = model.train(data='./dataset', epochs=3, imgsz=300)

• 결과 확인

model = YOLO('./runs/classify/train/weights/best.pt')
results = model.predict(source='./rock_sample.jpg', save=True)

이 글은 제로베이스 데이터 취업 스쿨의 강의 자료 일부를 발췌하여 작성되었습니다