人脸地标检测的光照问题

数据挖掘 图像识别 图像预处理 opencv
2022-03-07 15:03:07

我使用 ibug 和 yaleB 数据集训练了 dlib 人脸地标检测模型。当侧面出现亮光或阴影时,模型的精度会下降。训练时包括使用的默认参数和类似图像。有什么办法可以克服这个问题。

模型的样本结果

1个回答

也许你可以试试亮度数据增强;在 Keras 你有类似的东西:

ImageDataGenerator(brightness_range=[0.2,1.0])

如果您想根据火车图像生成 20% 到 100% 之间的不同亮度级别。例如(摘自Jason Brownlee的计算机视觉深度学习一书):

在此处输入图像描述