【python】去除图片高光

最新推荐文章于 2024-08-20 08:00:00 发布

DJames23

最新推荐文章于 2024-08-20 08:00:00 发布

阅读量7.2k

点赞数 4

分类专栏：图像处理 python 文章标签： opencv

本文链接：https://blog.csdn.net/DJames23/article/details/117358224

版权

OpenCV 高光去除像素聚类图像修复实时处理

关键词由CSDN通过智能技术生成

图像处理同时被 2 个专栏收录

2 篇文章

订阅专栏

python

2 篇文章

订阅专栏

该博客介绍了一种基于像素聚类的实时高光去除方法，并提供了使用OpenCV进行图像修复的Python实现。通过查找并修复图片中的高光部分，可以改善图像质量，适用于图像处理和计算机视觉领域。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考:

1.https://github.com/MarcioCerqueira/RealTimeSpecularHighlightRemoval

论文：Real-Time High-Quality Specular Highlight Removal using Efficient Pixel Clustering

2.图像增强：opencv去除图片的高光

import  cv2
import os,shutil
#找亮光位置
def create_mask(imgpath):
    image = cv2.imread(imgpath, cv2.IMREAD_GRAYSCALE)
    _, mask = cv2.threshold(image, 200, 255, cv2.THRESH_BINARY)
    return mask
#修复图片
def xiufu(imgpath,maskpath):
    src_ = cv2.imread(imgpath)
    mask = cv2.imread(maskpath, cv2.IMREAD_GRAYSCALE)
    #缩放因子(fx,fy)
    res_ = cv2.resize(src_,None,fx=0.6, fy=0.6, interpolation = cv2.INTER_CUBIC)
    mask = cv2.resize(mask,None,fx=0.6, fy=0.6, interpolation = cv2.INTER_CUBIC)
    dst = cv2.inpaint(res_, mask, 10, cv2.INPAINT_TELEA)
    return dst

if __name__=='__main__':
    rootpath = r"E:\data_set\img"
    masksavepath=r"E:\data_set\mask"
    savepath = r"E:\data_set\wcp\result"
    imgfiles = os.listdir(rootpath)
    for i in range(0, len(imgfiles)):
        path = os.path.join(rootpath, imgfiles[i])
        print(imgfiles[i])
        if os.path.isfile(path):
            if (imgfiles[i].endswith("jpg") or imgfiles[i].endswith("JPG")):
                maskpath =os.path.join(masksavepath, "mask_"+imgfiles[i])
                cv2.imwrite(maskpath, create_mask(path))
                dst=xiufu(path,maskpath)
                newname = 'xiufu_' + imgfiles[i].split(".")[0]
                cv2.imwrite(os.path.join(savepath, newname + ".jpg"), dst)
                shutil.copyfile(os.path.join(rootpath, imgfiles[i].split(".")[0] + ".xml"),
                                os.path.join(savepath, newname + ".xml"))