cv2可视化图片

self.original_image维度是3*h*w
       

 img=self.original_image.numpy()
        img = (img * 255).astype(np.uint8)

        # 由于tensor的形状是(3, H, W)，我们需要转置它以匹配OpenCV的(H, W, 3)格式
        tensor_np = np.transpose(img, (1, 2, 0))

        # 现在我们可以使用OpenCV的imshow函数来显示图像
        cv2.imshow('Image', tensor_np)
        # 等待按键关闭窗口
        cv2.waitKey(0)
        cv2.destroyAllWindows()


可视化的是BGR 不是RGB