【Pytorch】rgb转lab颜色空间转换

最近有需要，自己按照opencv实现的方式，自测！

个别tensor，numpy互转参考：https://oldpan.me/archives/pytorch-tensor-image-transform

颜色空间转换代码：

from __future__ import print_function
import torch
import numpy as np
from PIL import Image
import os
import cv2
from color_convertion import *
import osdef tensor_to_np(tensor):img = tensor.mul(255).byte()img = img.cpu().numpy().squeeze(0).transpose((1, 2, 0))return imgdef toTensor(img):assert type(img) == np.ndarray,'the img type is {}, but ndarry expected'.format(type(img))# img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)img = torch.from_numpy(img.transpose((2, 0, 1)))return img.float().div(255).unsqueeze(0)  # N x H x W X C numpy to tensor
def N_imgtoTensor(img):img = torch.from_numpy(img.transpose((0, 3, 1, 2)))return img.float().div(255)def N_tensor_to_np(tensor): #tensor value range is [0,1]img = tensor.mul(255).byte()         #tensor size [N,C,H,w]img = img.cpu().numpy().transpose((0,2,3,1))return img     #np_img size [N,H,w,C]def read_file(sn,tn):s = cv2.imread('some_data/'+sn+'.jpg')s = cv2.cvtColor(s,cv2.COLOR_BGR2LAB)t = cv2.imread('some_data/'+tn+'.jpg')t = cv2.cvtColor(t,cv2.COLOR_BGR2LAB)return s, tdef tensor_rgbtolab(sn,tn):s = cv2.imread('some_data/'+sn+'.jpg')s = cv2.cvtColor(s,cv2.COLOR_BGR2LAB)t = cv2.imread('some_data/'+tn+'.jpg')t = cv2.cvtColor(t,cv2.COLOR_BGR2LAB)s= toTensor(s)t=toTensor(t)return tensor2im(s),tensor2im(t)def tensor_original_img(sn,tn):s = cv2.imread('some_data/'+sn+'.jpg')t = cv2.imread('some_data/'+tn+'.jpg')s= toTensor(s)t=toTensor(t)return tensor_to_np(s),tensor_to_np(t)#1、 single img process!!!
s1,t1=read_file('p1','p2')s2,t2=tensor_rgbtolab('p1','p2')s3,t3=tensor_original_img('p1','p2')cv2.namedWindow("1",0)
cv2.namedWindow("2",0)
cv2.namedWindow("3",0)
cv2.resizeWindow("1",312,600)
cv2.resizeWindow("2",312,600)
cv2.resizeWindow("3",312,600)
cv2.imshow("1",np.vstack([s1,t1]))
cv2.imshow("2",np.vstack([s2,t2]))
cv2.imshow("3",np.vstack([s3,t3]))#2、multi batch to process!!!
def tensor_rgbtolab(sn,tn):s = cv2.imread('some_data/'+sn+'.jpg')t = cv2.imread('some_data/'+tn+'.jpg')temp=[]for i,img in enumerate([s,t]):img = cv2.cvtColor(img,cv2.COLOR_BGR2LAB)s=np.expand_dims(img,axis=0)temp.append(s)cat_result=np.concatenate(temp,axis=0)N_tensor=N_imgtoTensor(cat_result)print(N_tensor.shape)result=N_tensor_to_np(N_tensor)return resultoo=tensor_rgbtolab('p1','p2')
for i,img in enumerate(oo):cv2.imshow(str(i),img)cv2.waitKey(0)
cv2.destroyAllWindows()

【Pytorch】rgb转lab颜色空间转换相关推荐

RGB与Lab颜色空间互相转换
RGB与Lab颜色空间互相转换 1.Lab颜色空间同RGB颜色空间相比(见博客<光与色的故事--颜色空间浅析>),Lab是一种不常用的色彩空间.它是在1931年国际照明委员会(CIE)制 ...
RGB与Lab颜色空间互相转换持续更新中
RGB与Lab颜色空间互相转换 1.Lab颜色空间同RGB颜色空间相比(见博客<光与色的故事–颜色空间浅析>),Lab是一种不常用的色彩空间.它是在1931年国际照明委员会(CIE)制定 ...
Python实现RGB和Lab颜色空间互转
Python实现RGB和Lab颜色空间互转 https://github.com/rubund/debian-home-assistant/blob/1a3e8f7e4b9ddec60a4380e14 ...
RGB和CIELAB颜色空间转换及偏色检测
RGB转为CIELAB 首先RGB是不可以直接转为CIELAB颜色空间的,RGB需要先转为CIEXYZ颜色空间,然后再由CIEXYZ颜色空间转为CIELAB颜色空间.关于这2个颜色空间的互转,主要参考 ...
Python实现RGB转Lab颜色空间，PS：和PhotoShop转换结果一样
由于工作要求,需要把rgb颜色转成Lab颜色空间.网上一搜一大把的代码,但是和PS一对比,基本上都不对. 下面这张图是OpenCV官方文档的计算方法. 用Python实现代码如下: from math ...
Opencv的RGB到HSV颜色空间转换
从 RGB 到 HSL 或 HSV 的转换设 (r, g, b) 分别是一个颜色的红.绿和蓝坐标,它们的值是在 0 到 1 之间的实数.设 max 等价于 r, g 和 b 中的最大者.设 min ...
RGB与HSV颜色空间转换
RGB转化到HSV的算法: max=max(R,G,B): min=min(R,G,B): V=max(R,G,B): S=(max-min)/max: HSV颜色空间模型(圆锥模型) [2] if ...
RGB颜色空间与Lab颜色空间区别与联系(附转换代码)
RGB颜色空间 RGB颜色是红色(Red).绿色(Green)和蓝色(Blue)三基色的字母缩写.RGB色彩模式是通过三种基本颜色的不同程度的迭加来产生各种各样的不同颜色.这个标准能够涵盖人类视力所能 ...
MATLAB 将RGB颜色空间转为HSI颜色空间、LAB颜色空间原理及程序
一张彩色图像是由R.G.B三个通道组成,所以首先需要将彩色图像分为三个通道的图像. 以下是对猫图进行单通道提取得到的图像. HSI颜色空间 HSI[Hue-Saturation-Intensity(L ...

【Pytorch】rgb转lab颜色空间转换

【Pytorch】rgb转lab颜色空间转换相关推荐

最新文章

热门文章