[YOLO] Dataset 'data.yaml' error ❌ [Errno 2] No such file or directory

에러

model = YOLO('yolo11n-cls.pt')
model.train(data='data.yaml', epochs=100, batch=16, patience=5, name='type_classification')

 

위와 같이 작성했을 때 line 2 에서 아래와 같은 에러가 발생했다.

 Dataset 'data_yaml' error ❌ [Errno 2] No such file or directory: '(이상한 경로)/data_yaml/train'

 

 

data.yaml 파일의 경로가 이상하게 잡혀 있거나, data.yaml 파일이 제대로 생성되지 않은 줄 알았다.

해결

원래 Segmentation용으로 사용하려던 dataset을 그대로 classification으로 활용했는데, 이때 data='data.yaml' 과 같이 yaml 파일 형식으로 지정한 게 문제였다.

classification pretrained 모델인 yolo11n-cls.pt 를 사용하기 위해서는 아래와 같이 데이터셋을 지정해 줘야 한다.

# Train the model 'yolo11n-cls.pt'
results = model.train(data="mnist160", epochs=100, imgsz=64)

 

참고로 yolo11n-seg.pt 는 아래와 같다.

# Train the model 'yolo11n-seg.pt'
results = model.train(data="coco8-seg.yaml", epochs=100, imgsz=640)

 

 

Reference

https://docs.ultralytics.com/ko/tasks/classify/#models

분류

https://docs.ultralytics.com/ko/tasks/segment/#models

세그먼트