강의

멘토링

커뮤니티

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

mytrashcase님의 프로필 이미지
mytrashcase

작성한 질문수

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

inference된 이미지에서 마스크 영역만 jpg로 다시 저장할 수 있는지요?

작성

·

599

0

선생님 안녕하세요.

MMDetection 이용하여 inference 후,

마스크 이외 영역은 버리고, 마스크 영역만 검정색으로 원본 사이즈로 jpg로 다시 저장할 수 있을가요?

감사합니다.

답변 1

0

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

안녕하십니까, 

아래와 같이 해보시지요. 

# 원본 이미지 copy
img_copy = np.copy(img_arr_rgb)
# copy 된 이미지를 모두 검은색으로 설정. 
img_copy[:, :, :] = 0

# mask의 픽셀 위치 정보를 추출. 
x_array, y_array = np.where(results[1][0][0] == 1)

# 원본 이미지의 해당 픽셀 정보 값을 copy된 이미지로 복사. 
img_copy[x_array, y_array, :] = img_arr_rgb[x_array, y_array, :]
mytrashcase님의 프로필 이미지
mytrashcase

작성한 질문수

질문하기