질문 있습니다!

Question

섹션 10. Ultralytics Yolo 실습 - 02 강의를 따라 인크레더블 데이터 셋이 아닌, 제가 수집한 데이터로 학습을 수행중인데, 혹시 label을 한글로 했을 경우에 문제가 되는 것이 있을까 싶어 질문드립니다.

또 학습 진행 과정 중 밑과 같은 WARNING이 여러 줄 뜬 이후 별 문제 없이 학습이 진행되고있는데 어떤 오류인지 모르겠습니다..

아, 학습 완료된 모델로 detect를 할 때 output으로 box처리된 이미지 뿐만 아니라 text로도 출력을 하고싶은데, 이 또한 질문 드립니다!

항상 좋은 강의 감사드립니다.

train: WARNING: /content/food/Images/train/B070201_51763.jpg: ignoring corrupt image/label: could not convert string to float: 'None'

Answer

안녕하십니까,

1. 학습 시 label 값 자체를 한글로 할 수는 없습니다. 학습에 사용되는 모든 데이터는 숫자가 되어야 합니다.

다만 label 0 은 강아지, label 1은 고양이와 같이 매핑 테이블을 만든 다음에 label값을 매핑해주면 됩니다.

label_dict = {0:'강아지', 1:'고양이'}

2. warning은 큰 오류가 발생하지 않는 경우에 무시하셔도 될 것 같습니다.

3. 음,, 이건 bbox 좌표값과 class 예측 값을 python에서 문자열로 내용을 바꿔서, text 파일로 만드시면 될것 같습니다. python에서 문자열 값을 text 파일로 write하는 코드는 인터넷에서 쉽게 검색하실 수 있습니다.

감사합니다.