让我们首先使用 pip 安装这个库。

pip install imgaug

接下来,我们将使用 pip 命令在命令提示符下安装名为“IPyPlot”的 python 包:

pip install ipyplot

IPyPlot 是一个 Python 工具,允许在 Python Notebook 单元格中快速高效地显示图像。这个包将 IPython 与 HTML 相结合,以提供一种更快、更丰富、更具交互性的方式来显示图像。这个包的 ‘plot_images’ 命令将用于以网格状结构绘制所有图像。

此外,我们将导入扩充数据所需的所有必要包。

import imageio
import imgaug as ia
import imgaug.augmenters as iaa

增强的图像路径在此处定义。我们将使用鸟类图像作为示例。

input_img = imageio.imread(‘…/input/image-bird/bird.jpg’)

图像翻转

我们可以使用下面的命令水平和垂直翻转图像。以下代码中的“Fliplr”关键字水平翻转图像。同样,关键字“Flipud”垂直翻转图像。

#Horizontal Flip
 hflip= iaa.Fliplr(p=1.0)
 input_hf= hflip.augment_image(input_img)#Vertical Flip
 vflip= iaa.Flipud(p=1.0) 
 input_vf= vflip.augment_image(input_img)
 images_list=[input_img, input_hf, input_vf]
 labels = [‘Original’, ‘Horizontally flipped’, ‘Vertically flipped’]