深度学习必备-用Keras和直方图均衡化-数据增强

作者：王抒伟

编辑：王抒伟

本文100字3图，阅读约三分钟

算了

爱看多久看多久

在读这技术文章之前，请大家想象一个标准河南口音的娃在读这篇文章，那么你不知不觉，你的嘴角就上扬咯。

俺、遇到啥子问题撒~？

一

我现在写的文章都是因为遇到问题了，然后把解决过程给大家呈现出来

那么，现在我遇到了一个医学图像处理问题。

最近在处理医学图像的问题，发现DataSet一共只有400张图像,还是分为四类。

那怎么办呢？

可能你会说：这还不简单，迁移学习啊

soga,小伙子可以啊，不过今天我们不讲它（因为我还没实践过）

在这篇文章中，我们将讨论并解决此问题：

俺、咋滴解决的嘞~？

二

图像增强：它是什么？它为什么如此重要？

Keras：如何将它用于基本的图像增强。

直方图均衡化：这是什么？它有什么用处？

实现直方图均衡技术：修改keras.preprocessing image.py文件的一种方法。

俺、咋滴做的嘛~？

三

接下来我会从这四方面来讨论解决数据不足的问题

1.图像增强：它是啥（四声）？它为什么如此重要？

深度神经网络，尤其是卷积神经网络（CNN），尤其擅长图像分类任务。最先进的CNN甚至已经被证明超过了人类在图像识别方面的表现。

如果想克服收集数以千计的训练图像的高昂费用，图像增强则就是从现有数据集生成训练数据。

图像增强是将已经存在于训练数据集中的图像进行处理，并对其进行处理以创建相同图像的许多改变的版本。

这既提供了更多的图像来训练，也可以帮助我们的分类器暴露在更广泛的俩个都和色彩情况下，从而使我们的分类器更具有鲁棒性，以下是imgaug库中不同增强的一些示例

2.使用Keras进行基本图像增强

有很多方法来预处理图像，在这篇文章中，我借鉴使用keras深度学习库为增强图像提供的一些最常用的开箱即用方法，然后演示如何修改keras.preprocessing image.py文件以启用直方图均衡化方法。

我们将使用keras自带的cifar10数据集。但是，我们只会使用数据集中的猫和狗的图像，以便保持足够小的任务在CPU上执行。

加载 和 格式化数据

我们要做的第一件事就是加载cifar10数据集并格式化图像，为CNN做准备。

我们还会仔细查看一些图像，以确保数据已正确加载

先偷看一下长什么样？

cifar10图像只有32 x 32像素，所以在这里放大时看起来有颗粒感，但是CNN并不知道它有颗粒感，只能看到数据, 嗯，还是人类牛逼。

从ImageDataGenerator（）创建一个图像生成器

用keras增强 图像数据 非常简单。 Jason Brownlee 对此提供了一个很好的教程。

首先，我们需要通过调用ImageDataGenerator()函数来创建一个图像生成器，并将它传递给我们想要在图像上执行的变化的参数列表。

然后，我们将调用fit()我们的图像生成器的功能，这将逐批地应用到图像的变化。默认情况下，这些修改将被随机应用，所以并不是每一个图像都会被改变。大家也可以使用keras.preprocessing导出增强的图像文件到一个文件夹，以便建立一个巨大的数据集的改变图像，如果你想这样做，可以参考keras文档。

随机旋转图像

垂直翻转图像

备注：我感觉这里需要针对数据集，因为很少有人把狗翻过来看，或者拍照（hahhhh）

将图像垂直或水平移动20％

3.直方图均衡技术

直方图均衡化是指对比度较低的图像，并增加图像相对高低的对比度，以便在阴影中产生细微的差异，并创建较高的对比度图像。结果可能是惊人的，特别是对于灰度图像,如图

使用图像增强技术来提高图像的对比度，此方法有时也被称为“ 直方图拉伸

”，因为它们采用像素强度的分布和拉伸分布来适应更宽范围的值，从而增加图像的最亮部分和最暗部分之间的对比度水平。

直方图均衡

直方图均衡通过检测图像中像素密度的分布并将这些像素密度绘制在直方图上来增加图像的对比度。然后分析该直方图的分布，并且如果存在当前未被使用的像素亮度范围，则直方图被“拉伸”以覆盖这些范围，然后被“

反投影 ”到图像上以增加总体形象的对比

自适应均衡

自适应均衡与常规直方图均衡的不同之处在于计算几个不同的直方图，每个直方图对应于图像的不同部分;

然而，在其他无趣的部分有过度放大噪声的倾向。

下面的代码来自于sci-kit图像库的文档，并且已经被修改为在我们的cifar10数据集的第一个图像上执行上述三个增强。

首先，我们将从sci-kit图像（skimage）库中导入必要的模块，然后修改sci-kit图像文档中的代码以查看数据集第一幅图像上的增强

4.修改keras.preprocessing以启用直方图均衡技术。

以下是我们将要执行此功能的一般步骤：

在你自己的机器上找到keras.preprocessing image.py文件。

将image.py文件复制到您的文件或笔记本中。

为每个均衡技术添加一个属性到DataImageGenerator（）init函数。

将IF语句子句添加到random_transform方法，以便在我们调用时实现增强datagen.fit()。

对keras.preprocessing

文件进行修改的最简单方法之一就是将其内容复制并粘贴到我们的代码中。这将删除需要导入它。为了确保您抓取的是之前导入的文件的相同版本，最好抓取image.py您计算机上已有的文件。

运行将打印出机器上keras库的路径。路径（对于mac用户）可能如下所示：

这给了我们在本地机器上keras的路径。

继续前进，在那里导航，。.py文件。然后您可以将其内容复制到您的代码中。该文件很长，但对于初学者来说，这可能是最简单的方法之一。

编辑 image.py

在image.py的顶部，你可以注释掉这行：from ..import backend as K如果你已经包含在上面。

此时，请仔细检查以确保您正在导入必要的scikit-image模块，以便复制的模块image.py可以看到它们。

我们现在需要在ImageDataGenerator类的 __ init __

方法中添加六行代码，以便它具有三个代表我们要添加的增强类型的属性。下面的代码是从我目前的image.py中复制的。与#####侧面的线是我已经添加的线

该random_transform()（下）函数来响应我们一直传递到的参数ImageDataGenerator()功能。

如果我们已经设置了contrast_stretching，adaptive_equalization或者histogram_equalization参数True，当我们调用ImageDataGenerator()时（就像我们对其他图像增强一样）random_transform()将会应用所需的图像增强。

现在我们拥有所有必要的代码，并且可以调用ImageDataGenerator（）来执行我们的直方图修改技术。如果我们将所有三个值都设置为，则这是几张图片的样子True

训练并验证Keras CNN

最后一步是训练CNN并验证模型model.fit_generator()，以便在增强图像上训练和验证我们的神经网络.

四 End 总结

俺、写完（四声）唠！

四

我在这里展现了一张图片的增强结果，下图是我最后的增强结果

左上、增强测试图片 右上、增强结果

左下、原始数据标签 右下、原始数据

大家自己尝试一下哈，贵在实践，谁不会喊加油啊--->加油,加油，加油！

机器学习算法全栈工程师

一个用心的公众号

进群，学习，得帮助

你的关注，我们的热度，

我们一定给你学习最大的帮助