色卡定位及照片色彩校正

Dependencies

• opencv-python
• numpy

find_card.py

extract_color(color_card)

获取实际拍摄的照片中色卡各颜色色值

• param color_card: 透视校正完成的色卡图片
• return: 色卡中的颜色色值，以矩阵格式存储，color_matrix，shape: (3 , 24)

find_corner(img, blockSize=13, param_c=2, debug=False)

获取色卡四角的定位点

• param img: 输入图像
• param blockSize: Adaptive Threshold 参数 blockSize
• param param_c: Adaptive Threshold 参数 C
• param debug: 是否使用debug模式，输出各步骤结果和图片
• return: 四个角点坐标

get_color_card(img, points)

通过角点提取色卡部分，并对色卡进行透视校正，返回校正后色卡正视图

• param img: 输入图像
• param points: 色卡角点坐标
• return: 透视校正后的色卡图片

color_correction.py

recorrect_color(raw_img, A):

用系数矩阵A对图像进行颜色校正

• param raw_img: 原始图像
• param A: 系数矩阵
• return: 返回校正后的图像

card_verify.py

is_card_ok(color_card)

判断色卡提取是否正常，排除提取完成但提取错误情况

is_upsideDown(color_card)

判断色卡是否旋转

is_mirrored(color_card)

判断色卡是否镜像

is_upsideDown_and_mirrorred(color_card)

判断色卡是否镜像+旋转

Usage

提取单张照片中的色卡：

python3 find_card.py /WHERE/TO/FILE_PATH

批量提取文件夹中的色卡：

python3 find_card.py /WHERE/TO/DIR_PATH/

对单张照片校色

python3 color_correction.py /WHERE/TO/FILE_PATH