K.io.load_image读入数据转化到opencv中的数据

fluoxetine_drug

于 2025-03-12 14:31:14 发布

阅读量128

点赞数 3

文章标签： opencv 人工智能计算机视觉

本文链接：https://blog.youkuaiyun.com/fluoxetine_drug/article/details/146204569

版权

由于kornia读入的数据为3*W*H,而opencv中读入的数据为W*H*3

在二者中转化的可以做如下代码操作

import io

import kornia as K
import kornia.feature as KF
import matplotlib.pyplot as plt
import numpy as np
import torch
import cv2

src_1 = K.io.load_image("/root/code_2/APAP-Image-Stitching/images/demo1/bridge01.jpg", K.io.ImageLoadType.RGB32)

src_2 = src_1.permute(1,2,0).numpy()*256
src_2 = src_2[:, :, ::-1]
cv2.imwrite("test.jpg",src_2)

需要注意的是

kornia读取数据是0-1的颜色存储，并且为RGB，而opencv中的存储为，0-255，颜色通道为BGR。所以需要倒置。