在Colab中，使用&imagg&qot；进行图像数据增强并不能按预期工作 [英] In Colab doing image data augmentation with "imgaug" is not working as intended

问题描述

我正在增加我的图像数据集，它也包含关键点。为此，我使用imgaug库。以下是增强码：

kps = KeypointsOnImage(__keypoints, shape=_image.shape)

seq = iaa.Sequential([
iaa.Affine(
    scale={"x": (0.8, 1.2), "y": (0.8, 1.2)}, # scale images to 80-120% of their size, individually per axis
    translate_percent={"x": (-0.2, 0.2), "y": (-0.2, 0.2)}, # translate by -20 to +20 percent (per axis)
    rotate=(-90, 90), # rotate by -45 to +45 degrees
    order=[0, 1], # use nearest neighbour or bilinear interpolation (fast)
    cval=(0, 255),
),
iaa.Fliplr(0.5),
], random_order=True)


# Augment keypoints and images.
image_aug, kps_aug = seq(image = _image, keypoints=kps)

但在查看放大的图像时，我发现以下问题：

• 某些图像没有任何关键点。
• 在某些增强图像中，关键点在图像之外，尽管我一直在挡路上检查那些要保存在关键点不在图像内部的增强输出。

但奇怪的是，当我在我的PC上运行相同的代码时，它完全可以正常运行。但是当我在Google-Colab上运行它时，它会产生这些不需要的输出。为什么会发生这种情况？

推荐答案

我发现是版本问题。在Colab中，库imgaug附带了一个版本0.2.9，但是这个版本会产生这些不需要的输出。所以我卸载了这个现有版本，并安装了版本0.4.0。尽管在安装时显示以下错误：

ERROR: albumentations 0.1.12 has requirement imgaug<0.2.7,>=0.2.5, but you'll have imgaug 0.4.0 which is incompatible.

但我忽略了它，对我来说，它工作得很好。以下是卸载现有版本并安装所需版本的代码：

!pip uninstall imgaug
!pip install imgaug==0.4.0

我安装了该版本0.4.0，因为我也在本地PC上使用此版本，并且它在我身上工作没有任何问题。

这篇关于在Colab中，使用&imagg&qot；进行图像数据增强并不能按预期工作的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！