#imageio-rgb转opencv-bgr错误方式保存图片
cv2.imwrite("test.png",(data.detach().cpu().numpy()).astype(np.int8)[...,::-1])
#正确形式保存图片,特别注意是np.uint8类型不是np.int8
cv2.imwrite("test.png",(data.detach().cpu().numpy()).astype(np.uint8)[...,::-1])

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐