[OpenCV] error: OpenCV(4.5.4) /tmp/pip-req-build-sokf1_ny/opencv/modules/imgproc/src/resize.cpp:3929: error: (-215:Assertion failed) func != 0 in function 'resize'

출처 : https://medium.com/analytics-vidhya/images-manipulations-using-opencv-numpy-and-python-2e5538d35614

 

상황

np.argmax함수를 쓴 후 cv2.resize 해주려던 찰나 다음과 같은 에러 메세지를 마주하였다.

 

대략적인 코드

import numpy as np
import cv2

# np.argmax를 위한 과정
arg_max = np.zeros((512,512,3)) 
arg_max[...,1] = np.where(mask==1,1,0)
arg_max[...,2] = np.where(mask==2,1,0)
arg_max[...,0] = np.ones((512,512)) - arg_max[...,1] - arg_max[...,2]
arg_max = np.argmax(arg_max, axis=2)

# cv2.resize를 위한 과정
arg_max_resized = cv2.resize(arg_max, (128,128))

 

 

에러 메세지

error: OpenCV(4.5.4) /tmp/pip-req-build-sokf1_ny/opencv/modules/imgproc/src/resize.cpp:3929: error: (-215:Assertion failed) func != 0 in function 'resize'

 

원인

argmax 과정을 거치며 data type의 변형이 이루어졌기 때문이다. cv2.resize를 할 수 있는 적정한 data type인 np.uint8로변형해주어야 한다.

 

해결방법 

argmax 해준 변수 뒤에 .astype(np.uint8)을 붙여주면 된다.

 

 

수정된 코드

import numpy as np
import cv2

# np.argmax를 위한 과정
arg_max = np.zeros((512,512,3)) 
arg_max[...,1] = np.where(mask==1,1,0)
arg_max[...,2] = np.where(mask==2,1,0)
arg_max[...,0] = np.ones((512,512)) - arg_max[...,1] - arg_max[...,2]
arg_max = np.argmax(arg_max, axis=2).astype(np.uint8)

# cv2.resize를 위한 과정
arg_max_resized = cv2.resize(arg_max, (128,128))

 

잘 해결되었다.

 

 

 

-히비스서커스-