【图像处理实战】去除光照不均(Python)

这篇文章主要是对参考文章里面实现一种小拓展:

  • 可处理彩色图片(通过对 HSV 的 V 通道进行处理)
  • 本来想将嵌套循环改成矩阵运算的,但是太麻烦了,而且代码也不好理解,所以放弃了。

代码

python 复制代码
import cv2
import numpy as np


def light_compensate_hsv(src, grid_num):
    src_hsv = cv2.cvtColor(src, cv2.COLOR_BGR2HSV)
    src_v = src_hsv[:, :, 2]

    avg = np.mean(src_v)
    h, w = src_v.shape[:2]
    grid_row, grid_col = np.ceil(np.asarray([h, w]) / grid_num).astype(int)
    img_grid = np.zeros((grid_row, grid_col, 1), dtype=np.float32)

    for row in range(grid_row):
        for col in range(grid_col):
            img_roi = src_v[row * grid_num:                                           # row start
                            (row + 1) * grid_num if (row + 1) * grid_num < h else h,  # row end
                            col * grid_num:                                           # col start
                            (col + 1) * grid_num if (col + 1) * grid_num < w else w]  # col end

            img_grid[row, col] = np.mean(img_roi) - avg

    mask = cv2.resize(img_grid, (w, h), interpolation=cv2.INTER_CUBIC)
    mask = cv2.GaussianBlur(mask, (3, 3), 0)
    dst_v = np.uint8(np.float32(src_v) - mask)

    src_hsv[:, :, 2] = dst_v
    return cv2.cvtColor(src_hsv, cv2.COLOR_HSV2BGR)


def display(img):
    cv2.namedWindow('img', cv2.WINDOW_NORMAL)
    cv2.imshow('img', img)
    cv2.waitKey()


if __name__ == '__main__':
    image = cv2.imread(r'light_compensate_test_2.png')
    image_res = light_compensate_hsv(image, 16)

    res = np.hstack([image, image_res])
    display(res)

效果图

图也是用的参考文章里面的原图进行处理的。


最后一张是我自己的测试图,对于某些图来说效果比较好,可以用在文档阴影去除上。

参考文章

相关推荐
西柚小萌新32 分钟前
【Python爬虫基础篇】--4.Selenium入门详细教程
爬虫·python·selenium
起个破名想半天了37 分钟前
计算机视觉cv入门之答题卡自动批阅
人工智能·opencv·计算机视觉
鸿蒙布道师1 小时前
OpenAI为何觊觎Chrome?AI时代浏览器争夺战背后的深层逻辑
前端·人工智能·chrome·深度学习·opencv·自然语言处理·chatgpt
橘猫云计算机设计1 小时前
springboot基于hadoop的酷狗音乐爬虫大数据分析可视化系统(源码+lw+部署文档+讲解),源码可白嫖!
数据库·hadoop·spring boot·爬虫·python·数据分析·毕业设计
YOULANSHENGMENG1 小时前
linux 下python 调用c++的动态库的方法
c++·python
神奇侠20241 小时前
基于opencv和PaddleOCR识别身份证信息
opencv·paddleocr
SsummerC1 小时前
【leetcode100】零钱兑换Ⅱ
数据结构·python·算法·leetcode·动态规划
一眼青苔2 小时前
切割PDF使用python,库PyPDF2
服务器·python·pdf
电商数据girl2 小时前
产品经理对于电商接口的梳理||电商接口文档梳理与接入
大数据·数据库·python·自动化·产品经理
三道杠卷胡3 小时前
【AI News | 20250424】每日AI进展
人工智能·pytorch·python·语言模型·github