활연개랑

[tensorflow] tf.numpy_function() dim error 문제해결 본문

Python

[tensorflow] tf.numpy_function() dim error 문제해결

승해tmdhey 2022. 5. 8. 19:10
반응형

tf.numpy_function을 사용할 때 dim 관련 오류가 날 수 있습니다.

tf.numpy_function을 이미지에 사용할 경우 reshape이 필요하기 때문에 다음과 같이 reshape을 해주어야 합니다. 

image_shape = tf.shape(image)
image = tf.numpy_function(definition(func), [image], tf.float32)
image = tf.reshape(image, image_shape)

이렇게 reshape을 해주면 dimensipn오류를 해결할 수 있습니다.

 

 

'Python' 카테고리의 다른 글

[numpy(넘파이)] python numpy array이미지 dtype변경 - float to uint  (0) 2022.05.25
[ERROR 해결] Input 0 of layer "conv2d" is incompatible with the layer: expected min_ndim=4, found ndim=2. Full shape received  (0) 2022.05.09
[tensorflow] Input 0 of layer "conv2d" is incompatible with the layer: expected min_ndim=4, found ndim=2. Full shape received: (300, 400)  (0) 2022.05.04
[OpenCV] cv2.error: OpenCV(4.5.5) /io/opencv/modules/imgproc/src/histogram.cpp:3439: error: (-215:Assertion failed) _src.type() == CV_8UC1 in function  (0) 2022.04.29
[opencv] cv2.error: OpenCV(4.5.5) /io/opencv/modules/imgproc/src/histogram.cpp:3439: error: (-215:Assertion failed) _src.type() == CV_8UC1 in function 'equalizeHist'stackoverflow.com/questions/23660929/how-to-check-whether-a-jpeg-image-is-color-or-gray-..  (0) 2022.04.26
'Python' Related Articles more