В Colab увеличение данных изображения с помощью imgaug не работает должным образом

Я расширяю свой набор данных изображения, который также содержит ключевые моменты. По этой причине я использую 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, он создает эти нежелательные результаты. Почему это происходит?


object-detection computer-vision google-colaboratory data-augmentation keypoint
person hafiz031    schedule 25.06.2020    source источник

Ответы (2)


arrow_upward
1
arrow_downward

Я обнаружил, что это проблема с версией. В 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, так как на моем компьютере я работал с этой версией, и она у меня сработала без проблем.

person hafiz031    schedule 25.06.2020

arrow_upward
0
arrow_downward

Colab поставляется с версией imgaug 0.2.9, но эта версия не соответствует требованиям к альбументации. Я столкнулся с той же проблемой, использовал эту команду для исправления ошибки:

!pip uninstall imgaug && pip uninstall albumentations && pip install git+https://github.com/aleju/imgaug.git
person Priya Balaji    schedule 04.07.2020