인프런 커뮤니티 질문&답변

비밀님의 프로필 이미지
비밀

작성한 질문수

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

mask r-cnn custom data로 학습 하려다가 발생하는 오류 문의드립니다

작성

·

295

0

안녕하세요. 저는 현재 제 custom data로 mask rcnn을 통해 학습을 진행하고자 하고 있습니다.
 
현재 각 label을 1로 masking한 후, 강의처럼 폴더를 만들어서 진행하였습니다.
그림에는 image가 추가가 안되어 있지만 이후에 추가를 하여서 데이터셋 자체는 잘 구축하였습니다. 각 image당 mask 이미지가 24장 있는 dataset입니다. 이미지 데이터는 이미지와 mask 모두 (512,512) 크기입니다
이후에 순차적으로 coco dataset으로 만들어주고, 코드를 실행하다가 마지막 train_detctor 부분의 에러를 해결하지 못하고 있습니다..
 
실행을하고자 하는데 에러가 떠서 그냥 에러 전체를 첨부해보겠습니다. type이 현재 float 형식으로 되어 있는 것 같은데 어디에서 잘못된건지 잘 모르겠어서 이렇게 문의합니다. 감사합니다.
------------------------------------------------------------------------------------------------------------------------

답변 2

0

비밀님의 프로필 이미지
비밀
질문자

강의를 다시 보며 제 실수를 찾았습니다. 좋은 강의 감사합니다! 

0

권 철민님의 프로필 이미지
권 철민
지식공유자

안녕하십니까, 

정확하지는 않지만, masking 정보를 polygon 형태로 변환하는 과정에서 잘못 변환된 정보가 있는 것 같습니다. 

보시면 poly2mask 에서 오류가 나옵니다.  강의 중 Kaggle Nucleus Segmentation 실습 - Mask 이미지 파일에서 Polygon 좌표 추출하기 영상을 참조하셔서 polygon 정보들을 다시 한번 확인해 보시기 바랍니다. 

비밀님의 프로필 이미지
비밀

작성한 질문수

질문하기