inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

Mask-RCNN image segmentation 질문

212

노윤호

작성한 질문수 3

0

안녕하세요 강사님, 좋은 강의 정말 잘 듣고 있습니다.

Mask-RCNN이 instance segmentation이기 때문에 검출된 객체가 같은 class라도 그 영역이 다른 색으로 표시되는데요,

혹시 같은 class일때는 동일한 색으로 표시되게도 설정할 수 있을까요?

감사합니다.

python tensorflow keras 컴퓨터-비전 머신러닝 배워볼래요? 딥러닝

답변 1

0

권 철민

안녕하십니까, 

잘 듣고 계시다니 저도 기분이 좋군요. 

같은 class일때 동일한 색으로 표시하게 할 수 있습니다. 

Pascal VOC 데이터를 이용한 Train 실습 - Train된 모델을 이용하여 이미지와 Video inference 실행 강의 예제는 동일한 클래스의 경우 같은 색으로 표시하도록 실습 코드가 되어 있습니다. 

해당 소스코드를 보면 def get_detected_img() 함수에서 아래와 같은 부분으로 동일한 class id를 가지는 경우 동일한 색으로 표시하게 되어 있습니다. 

color_index = result_ind % len(colors)

color = colors[color_index]

 

opencv inference 예제나 다른 예제의 경우에도 이를 응용하면 됩니다. 

opencv의 경우 시각화 코드에서 아래와 같이 colorIndex를 random 하게 가져오는데 이걸 위 코드를 참조하여,  class id별로 동일하게 적용하는 로직으로 변경하시면 됩니다.  

colorIndex = np.random.randint(0, len(colors)-1)

color = colors[colorIndex]

 

감사합니다. 

MMDetection 버전 이슈

0

51

2

강의 환경설정 질문

0

61

2

Custom Dataset에서의 polygon 정보 관련

0

113

3

cvat.ai 보안 수준이 궁금합니다

0

98

2

캐클 nucleus 챌린지 runpod 실습 코드 에러 질문드립니다.

0

120

3

추론 결과의 Precision(또는 mAP) 평가 방법

0

94

2

mmdetection mask rcnn inferenct 실습 시 runpod 템플릿 관해서 질문드립니다.

0

68

2

runpod에서 google drive 연결 시 오류 발생

0

126

2

로드맵 선택

0

74

1

mmcv

0

65

2

Anchor box의 Positive 처리 위치

0

71

2

해당 강의 runpod 적용 후 에러 제보드립니다

0

95

2

run pod credit 관련 제보

0

125

2

mmdetection 2.x과 3.x 호환 관련 표기

0

89

2

mm_faster_rcnn_train_kitti.ipynb 실행 오류

0

114

3

질문 드립니다.

0

89

3

mm_faster_rcnn_train_coco_bccd 실행 오류 질문드립니다.

0

89

1

강사님께 수정을 제안드리고 싶은 것이 있습니다.

0

102

1

google automl efficientdet 다운로드 및 설치 오류

0

87

1

이상 탐지에 사용할 비전 기술 조언 부탁드립니다.

0

111

2

OpenCV 관련 질문드립니다.

0

87

2

mmcv 설치관련해서 문의드려요

0

357

3

강의 구성 관련해서 질문이 있습니다

1

141

2

모델 변환 성능 질문드립니다.

0

125

1