yolo5 custom data training

안녕하세요!

cvat로 어노테이션을 진행한뒤 yolo 5 로 학습할대 에러가 나와서 문의 드립니다.

학습을 한 셀에서 결과가 ....

20 epochs completed in 0.033 hours.

Optimizer stripped from content/dataset/result/plt/weights/last.pt, 92.8MB Optimizer stripped from content/dataset/result/plt/weights/best.pt, 92.8MB Validating

content/dataset/result/plt/weights/best.pt...

Fusing layers...

Model summary: 267 layers, 46129818 parameters, 0 gradients, 107.7 GFLOPs

Class Images Instances P R mAP50 mAP50-95: 0% 0/8 [00:00<?, ?it/s]

Exception in thread Thread-8: Traceback (most recent call last):

File "/usr/lib/python3.8/threading.py", line 932, in bootstrapinner self.run() File "/usr/lib/python3.8/threading.py", line 870, in run self._target(*self._args, **self._kwargs) File "/content/yolov5/utils/plots.py", line 287, in plot_images cls = names[cls] if names else cls KeyError: 0

이후에...

Exception in thread Thread-10:

Exception in thread Thread-12:

에러가 납니다.

어떤 문제일까요??;

- 학습 관련 질문을 남겨주세요. 상세히 작성하면 더 좋아요!
- 먼저 유사한 질문이 있었는지 검색해보세요.
- 강의 내용을 질문할 경우 몇분 몇초의 내용에 대한 것인지 반드시 기재 부탁드립니다.
- 서로 예의를 지키며 존중하는 문화를 만들어가요.
- 잠깐! 인프런 서비스 운영 관련 문의는 1:1 문의하기를 이용해주세요.