-
카테고리
-
세부 분야
컴퓨터 비전
-
해결 여부
미해결
mask r-cnn custom data로 학습 하려다가 발생하는 오류 문의드립니다
22.01.01 04:25 작성 조회수 237
0
안녕하세요. 저는 현재 제 custom data로 mask rcnn을 통해 학습을 진행하고자 하고 있습니다.
현재 각 label을 1로 masking한 후, 강의처럼 폴더를 만들어서 진행하였습니다.
그림에는 image가 추가가 안되어 있지만 이후에 추가를 하여서 데이터셋 자체는 잘 구축하였습니다. 각 image당 mask 이미지가 24장 있는 dataset입니다. 이미지 데이터는 이미지와 mask 모두 (512,512) 크기입니다
이후에 순차적으로 coco dataset으로 만들어주고, 코드를 실행하다가 마지막 train_detctor 부분의 에러를 해결하지 못하고 있습니다..
실행을하고자 하는데 에러가 떠서 그냥 에러 전체를 첨부해보겠습니다. type이 현재 float 형식으로 되어 있는 것 같은데 어디에서 잘못된건지 잘 모르겠어서 이렇게 문의합니다. 감사합니다.
------------------------------------------------------------------------------------------------------------------------
답변을 작성해보세요.
0
0
권 철민
지식공유자2022.01.01
안녕하십니까,
정확하지는 않지만, masking 정보를 polygon 형태로 변환하는 과정에서 잘못 변환된 정보가 있는 것 같습니다.
보시면 poly2mask 에서 오류가 나옵니다. 강의 중 Kaggle Nucleus Segmentation 실습 - Mask 이미지 파일에서 Polygon 좌표 추출하기 영상을 참조하셔서 polygon 정보들을 다시 한번 확인해 보시기 바랍니다.
답변 2