[YOLOv5] 커스텀 데이터 roboflow 라벨링 - epochs 5,50(실패)

 

 

YOLO v5 PyTorch를 선택하고, zip으로 데이터를 꺼낸다음

 

이 경로에 'test01_custom'이라는 폴더를 만들어 그 안에 압축 해제를 해주었다.

 

여기서 오류가 발생했었는데,

# 해놓은 경로 대신

train 이미지 파일이 있는 경로를 직접 복사하여 붙여넣으니 해결되었다.

train\images가 있는 경로. 위쪽 경로 오른쪽 마우스 -> 경로 복사하기 후 붙여넣기 하였음

 

 

 

python train.py --img 640 --batch 16 --epochs 50 --data test01.yaml --cfg models/yolov5s.yaml --weights yolov5s.pt --cache

 

 

으로 학습을 진행하다가, 너무 시간이 길어져 epochs를 5, 50으로 했는데 정상적으로 동작하지 않았다. 

 

val_batch0_labels는 검증 데이터셋의 첫 번째 배치에 대한 ground truth 레이블을 시각화한 이미지인데,

즉 객체의 위치와 클래스 정보를 표시한 정답 레이블이다.

val_batch0_pred는 검증 데이터셋의 첫 번째 배치에 대한 모델 예측 결과를 시각화한 이미지이며,

모델이 학습을 통해 객체를 얼마나 잘 감지하고 분류하는지 보여준다.

그런데 난 val_batch0_pred 이미지에 아무 표시도 없었고, 그것은 모델이 객체를 감지하지 못한 것이다,,.

이런 경우 학습이 제대로 이루어지지 않았을 가능성이 있다.

 

그래서 원래대로 epochs를 300으로 변경하고 다시 진행해보았다.

 

(다음 글에서 계속...

스포: 300으로 한거 코랩에서 됐음

파이참에서는 300번 돌아가는 속도도 매우 느리고 결과도 안나왔음.. 왜일까?)