让我们首先使用 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’]