1 | initial version |
cv2.imshow() does not support 2 channel images. you added some weird code there, which probably just should be deleted.
cv2.imwrite() does not support float images, you'll have to convert to np.uint8 (and probably scale it)