sung-chul lee

휴~~작가님.^^
coco 데이터 셋과 비교해보고 , 로그도 찍어보고 하다가 문제점을 발견하고 이제서야 해결했습니다.^^
load_mask메서드를&nbsp; coco 데이터 셋과 비교해서 category_id를 만드는 작업을 했습니다.
막 두서없이 질문하는데 답변해 주셔서 감사합니다.
최소한 데이터 셋을 어떻게 구성해야 하는지는 정확히 알게 되었네요.^^;;;
학습 결과로 segmentation 진행해 봤는데 성공적으로 되었습니다.^^;;;
도움을 주셔서 정말 감사드립니다

matterport mask rcnn의 image_id 와 class_id의 연결

수정을 해서 테스트를 해보니
model.py&nbsp;&nbsp;load_image_gt 메소드의&nbsp;
source_class_ids = dataset.source_class_ids[dataset.image_info[image_id]["source"]] 에서 에러가 나고 있는데요
&nbsp;dataset.source_class_ids 에는 현재 {'': [0], 'AL': [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]}&nbsp; 이렇게 들어가 있고요
dataset.image_info[image_id]["source"] 의 값은 단일값 (예.&nbsp; 1 혹은 3 ) 과 같은 값이 들어오고 있습니다.
그러다보니 여기에서 에러가 나고 있고요..
사전에 클래스 등록은&nbsp;
&nbsp; self.add_class("AL", 1, "1")
&nbsp; &nbsp; &nbsp; &nbsp; self.add_class("AL", 2, "2")
&nbsp; &nbsp; &nbsp; &nbsp; self.add_class("AL", 3, "3")
&nbsp; &nbsp; &nbsp; &nbsp; self.add_class("AL", 4, "4")
&nbsp; &nbsp; &nbsp; &nbsp; self.add_class("AL", 5, "5")
&nbsp; &nbsp; &nbsp; &nbsp; self.add_class("AL", 6, "6")
&nbsp; &nbsp; &nbsp; &nbsp; self.add_class("AL", 7, "7")
&nbsp; &nbsp; &nbsp; &nbsp; self.add_class("AL", 8, "8")
&nbsp; &nbsp; &nbsp; &nbsp; self.add_class("AL", 9, "9")
&nbsp; &nbsp; &nbsp; &nbsp; self.add_class("AL", 10, "0")
이렇게 하고&nbsp;
&nbsp;self.add_image(
&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; category_name,
&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; image_id=a['filename'],&nbsp; # use file name as a unique image id
&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; path=image_path,
&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; width=width, height=height,
&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; polygons=polygons
&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; &nbsp; )
여기에서 category_name에 클래스 명을 넣어주고 있습니다.
category_name 이 "1","2"...이와 같습니다.
귀찮게 해서 정말 죄송합니다.. 도움을 부탁드립니다..
괜찮으시면 정말 찾아뵙고 도움을 부탁드리고 싶네요.ㅜㅜ

제가 작가(^^-어디에선가 작가님이라 불리시는 것을 좋아하신다고 되어 있는것 같아서..^^)님의 설명을 잘 못 이해해서 많이 돌고 있군요.ㅎㅎ.다시 해보겠습니다.ㅎㅎㅎ

아,,소스 상의 comment가 add_class 시에 source로 되어 있던데, source가 아니라 그냥 클래스 명을 넣어주면 되는 건가요? ^^;;;아직 삽질 중입니다.ㅎㅎㅎ

계속 질문을 하게 되네요.^^;;
BallonDataset의 경우 class가 balllon 하나 뿐이지 않나요?
그래서 add_class 후에 add_image를 할 때 입력하는 값이 source이지 않나요?
그래서 coco dataset의 경우 그냥 coco로 입력하는 것으로 보였는데요
coco dataset의 경우는 class 가 80여 가지일텐데요..소스를 보다보니 coco dataset의 경우
load_mask 에서 category_id로&nbsp;
연결하는것 같다는 생각이 들었습니다.(솔직히 맞는지는 잘 모르겠습니다)
coco dataset처럼 여러 종류를 가진 이미지들을 train 시키고 싶은건데요
계속 class_id와 image_id가 어떻게 연결되는지를 몰라 혼돈 중에 있습니다.^^;;
하나의 이미지는 train시켜서 detection 했는데요, 여러가지 하려니 막히네요.ㅎㅎㅎ
도움을 부탁드립니다.

아, 그렇다면 여러개의 클래스에 대해서는 먼저 add_class를 통해 클래스를 등록하고, add_image시 클래스명을 이미지에 따라 등록되게 해주면 이미지와 클래스가 연결이 되겠군요..?
맞게 이해한 건가요? ^^;;
그런데, 정말 상세한 설명 감사드립니다. ^^
강의를 만들어 주신것만도 정말 감사드리는데,&nbsp; 문의에 대한 답변도 정말 잘 해주셔서 감사합니다.

명확한 답변 감사합니다.^^

sung-chul lee

Posts