cv2.error: OpenCV(4.2.0) C:\projects\opencv-python\opencv\modules\imgcodecs\src\loadsave.cpp:668: error: (-215:Assertion failed) image.channels() == 1 || image.channels() == 3 || image.channels() == 4 in function 'cv::imwrite_'
使用CV2.imread或者CV2.imwrite的时候一定要注意ndarray.shape() 返回的是channel_first 还是channel_last.
opencv只支持channel_last, 也就是说只支持(112L,112L,3L)而如果输入shape为(3L,112L,112L)就会报这个错误!!!
所以一定要当心是否做过np.transpose操作来转换适应不同的模型,因为model的输入一般是channel_first.
解决方案:transpose回去!!!
aligned = np.transpose(aligned, (1, 2, 0)) #have to be careful about the transpose。
如果transpose错了,你的图片可以被旋转或者flip