ValueError: could not broadcast input array from shape… 에러 해결 방법

GAN 을 이용하여 이미지를 합성 하다 보면 could not broadcast input array from shape could not broadcast input array from shape 에러가 발생하는 경우가 종종 발생한다.

 

ValueError: could not broadcast input array from shape (64,64,3) into shape (64,64)

 

이런 경우는 어떻게 해결 해야 할까?

 

이런 문제는 대부분 Source 이미지가 흑백 이미지라서 발생하는 경우가 많다. 아래의 코드처럼 Pillow 라이브러리를 이용하여 RGB 로 다시 저장을 하면 이런 문제는 쉽게 해결이 가능하다.

 

from PIL import Image
img = Image.open(img_name).convert('RGB')
img.save(image_name)