在 Colab 中使用“imgaug”进行图像数据增强并没有按预期工作

2024-02-25

我正在扩充我的图像数据集,其中也包含关键点。因此我正在使用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)

但在查看增强图像时,我发现了以下问题:

  • 有些图像没有任何关键点。
  • 在某些增强图像中,关键点位于图像之外,尽管我不断检查以阻止将那些增强输出保存在关键点不在图像内部的位置。

但奇怪的是,当我在我的电脑上运行相同的代码时,它运行完全正常。但是当我在 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因为我也在我的本地电脑上使用这个版本,它对我来说没有任何问题。

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

在 Colab 中使用“imgaug”进行图像数据增强并没有按预期工作 的相关文章

随机推荐