• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

yolo 커스텀 데이터 학습시 범위지정

21.09.05 14:09 작성 조회수 413

0

yolo를 활용해서 통신주를 찾으려고 합니다.(여러번 질문드려 죄송합니다;;)

전에 조언해주신대로 yolo v5로 변경하고, 학습데이터수도 100장이상이 되도록 수정중에 있습니다.

다만, CVAT툴로 영역지정시 아래그림과 같이 깔끔하게 하늘배경에 통신주가 있는경우도 있지만,

풀숲이나, 기타 장애물로 가려진 경우도 다수입니다. 

이런경우 영역지정을 잘나온부분만 특정(상단 안테나, 하단 통신장치)해서 학습시키는 것이 맞을지 그래도 전체(전봇대포함전체)를 지정해주는 것이 맞을지 궁금합니다.

(배경과 상관없이 선택된 영역들의 공통부분을 알아서 인지하는 알고리즘?) 

다른 질문이긴 합니다만

yolov3_inference.ipynb 파일을 참고하면 weights yolov3.pt 파일을 다운받아 

바로 detect하면 coco모델의 사람, 차 등이 바로 인지가 되는데, 

커스텀 데이터로 학습후 detect시 기존 인지되는 물체에 커스텀객체가 추가로 인지되게 할 수 있을까요?

best.pt결과에 기존 coco모델(80개)+내가 커스텀(인크레더블) 같이 표시

다시한번 좋은 강의에 감사드립니다^^

답변 1

답변을 작성해보세요.

0

안녕하십니까, 

1.  장애물로 가려져 있더라도 통신주 전체가 있는게 더 좋습니다. 특히나 테스트 하려는 데이터가 장애물등으로 일부 가려져 있는 경우라면, 더더욱 장애물로 가려져 있는 통신주 전체를 학습 데이터로 해 주시는 게 좋습니다. 

2. coco pretrained 모델을 이용하더라도 기존 coco 80개 + 커스텀(인크레더블)을 구성하기는 어렵습니다. 

왜냐하면 몇개의 class를 학습할지 미리 갯수를 정해서 설정했기 때문입니다. 방법이 아예 없는것은 아닙니다. 학습 시 coco dataset과 인크레더블 데이터를 섞습니다. coco dataset의 경우 pretrained 되어 있기 때문에, 데이터가 많이 필요하지는 않습니다. 대신 인크레더블 데이터는 많이 필요합니다. 인크레더블 데이터는 사람으로 나올 수 있으므로  상대적으로 매우 많은 데이터 세트가 필요합니다. 그렇게 학습 데이터 세트를 만들고 class 개수를 81개 설정한 뒤 다시 학습하면 됩니다. 

감사합니다.