• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

mask-rcnn에서 dataset 준비

21.08.10 16:19 작성 조회수 254

0

강사님 안녕하세요, 정말 좋은강의 감사드립니다.

제가 최종적으로 구현하고 싶은 모델은 이미지 상에 존재하는 다양한 모양의 원형 입자를 구분하고, 이 좌표를 추출하는 것입니다.

이를 위해 mask-rcnn을 적용해보고자 하는데, 하나의 문제점이 제가 구분하고자 하는 모양의 내부에 구멍이 뚫려 있다는 것입니다.

예를 들면 위와 같이 내부 구멍의 모양으로 구분이 되게 되는데, 따라서 segmentation 좌표가 외부1, 내부4로 나오게 됩니다.

외부 segmentation 좌표에서 내부 segmentation 좌표를 빼는 방법으로 위와 같이 마스크를 구현을 해보긴 했으나,  이를 어떻게 coco dataset으로 변경해서 학습해야할까요? 구글에서도 정보를 찾기 어려워 이렇게 질문을 남기게 되었습니다.

항상 감사드립니다.

답변 1

답변을 작성해보세요.

0

안녕하십니까, 

음,, 어제부터 저도 계속 알아보고 있는 중인데, 언급하신 masking을 coco segmentation인 polygon 형태로 변환할 수 있는지 의문입니다. contour를 사용해서 내부 masking을 coco segmetation polygon 형태로 표현이 가능할지 잘 모르겠습니다.(뇌피셜로는 안될 것 같습니다만)

mmdetection에서 coco segmentation이 아니고 다른 Masking Dataset으로 시도는 해볼수 있겠는데, Tutorial 이 없어서 소스코드를 뜯어볼수 밖에 없는데 시간이 많이 소모 될 것 같습니다. 

일단 데이터와 소스코드를 colab 형태로 만들어 주시면 제가 시간나는대로 함 시도해 보겠습니다. 

감사합니다.