axes don't match array , im[..., ::-1].transpose((0, 3, 1, 2)) # BGR to RGB, BHWC to BCHW, (n, 3, h, w) Additional No response fromskimage.colorimportgray2rgbcvImage=gray2rgb(cvImage)cvImage=cv2.cvtColor(cvImage,cv2.COLOR_GRAY2BGR) ...