关键词:消除不同图像数据色调差异/消除颜色深浅差异/病理/机器学习/深度学习/人工智能

机器学习训练图像数据时可能会因为图像数据之间的颜色深浅,色调等影响训练和预测结果,本方法可以减弱颜色深浅/色调不同对训练造成的影响

本栗子用到的是彩色图片

import numpy as np
import cv2
import histomicstk as htk

root = 'list.txt'
infer_path = './abc.png'
infer = cv2.imread(infer_path)
for path in open(root):
    path = path.replace('\n', '')
    print('>>>',path)
    input_img = cv2.imread(path)
    meanRef, stdRef = htk.preprocessing.color_conversion.lab_mean_std(infer)
    imNmzd = htk.preprocessing.color_normalization.reinhard(input_img, meanRef, stdRef)
    cv2.imwrite(path, imNmzd)

  其中 histomicstk为某个库,可使用pip install histomicstk安装,root为目标文件遍历的列表(可参考我写的上一篇笔记),infer_path为标准化图像(用人眼观察主观得到的感觉颜色/色调等都不错的图片)