import cv2
import os.path
import glob
import numpy as np
from PIL import Image
def convertPNG(pngfile,outdir):
# READ THE DEPTH
im_depth = cv2.imread(pngfile)
#apply colormap on deoth image(image must be converted to 8-bit per pixel first)
im_color=cv2.applyColorMap(cv2.convertScaleAbs(im_depth,alpha=15),cv2.COLORMAP_JET)
#convert to mat png
im=Image.fromarray(im_color)
#save image
im.save(os.path.join(outdir,os.path.basename(pngfile)))
convertPNG('./source/1.png','./source/output')
把convertPNG中的两个参数,分别换成输入图片路径和输出文件夹路径即可。
参考:
【1】 https://www.cnblogs.com/BambooEatPanda/p/9921446.html