import cv2
import numpy as np
import os
file = "/home/imed/下载/transunet_pytorch-main/results/label/batch_10_48.tif"
image = cv2.imread(file)
print(image.shape) #查看图像的通道数
print(np.unique(image)) #查看图像的格式,是否二值化(若只有0和255则是二值化,若是0和255之间的像素值,则未被二值化)
flag = image.copy()
flag = cv2.cvtColor(flag, cv2.COLOR_RGB2GRAY) #将RGB三通道改为单通道,通过.shape查看;
thres, flag = cv2.threshold(flag, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU) # 二值化处理,只留下0和255(0是黑色,255是白色)
w, h, c = image.shape
for i in range(w):
for j in range(h):
if flag[i, j] > 0:
image[i, j, :] = [0, 0, 255] #将二值化后像素值大于零(也就是255白色调为红色(0,0,255))
else:
image[i, j, :] = [255,255,255] #将二值化后像素值等于零(也就是0黑色调为白色(255,255,255)),注意是BGR,顺序颠倒了
cv2.imwrite(os.path.join("/home/imed/下载/transunet_pytorch-main/results/colored",os.path.basename(file)), image)
PPT汇报更改图像背景色及血管颜色(代码实现)
最新推荐文章于 2023-07-08 18:59:16 发布