用python将单通道变成三通道，并将深度从Uint8变为Uint16重命名到其他文件中

最新推荐文章于 2023-11-04 21:20:04 发布

ojbko

最新推荐文章于 2023-11-04 21:20:04 发布

阅读量1.2k

点赞数 2

本文链接：https://blog.csdn.net/ojbko/article/details/108658001

版权

代码如下：

import cv2
import numpy as np
import os
import numpy as np
import matplotlib.image as mp
from skimage import img_as_ubyte
from PIL import Image

path = "/home/jrx/jiangruixiang/planerecover-master/planerecover-master_nyu/nyu/00/"  
savepath = "/home/jrx/jiangruixiang/planerecover-master/planerecover-master_nyu/new_nyu/"  
filelist = os.listdir(path) 
total_num = len(filelist)  
for i in range(total_num):
    jpg_name = path + format(str(i), '0>6s') + '_depth.png' 
    image = cv2.imread(jpg_name)
    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    img2 = np.zeros_like(image)
    img2[:,:,0] = gray
    img2[:,:,1] = gray
    img2[:,:,2] = gray
    img2=img2.astype(np.uint16)   #uint8变成uint16
    savejpg_name = savepath + format(str(i), '0>6s') + '_depth.png' 
    cv2.imwrite(savepath +format(str(i), '0>6s') + '_depth.png' , img2)