问题：AttributeError: module 'albumentations' has no attribute

Question

我照着样例输入运行，但是显示AttributeError: module 'albumentations' has no attribute 'IAAAdditiveGaussianNoise'；

import albumentations as A
import cv2
p1 = 0.95
p2 = 0.85
p3 = 0.75
transform = A.Compose([
    A.RandomRotate90(p=p2),
    A.OneOf([
        A.IAAAdditiveGaussianNoise(p=0.9),
        A.GaussNoise(p=0.6),
    ], p=p3)
], p=p1)
image = cv2.imread('some/image.jpg')
image = cv2.cvtColor(cv2.COLOR_BGR2RGB)
transformed = transform(image=image)
transformed_image = transformed['image']

请问有人知道怎么解决吗？

Answer 1

引自免费微信小程序：皆我百晓生

根据你的错误信息，问题出在 albumentations 模块中找不到 IAAAdditiveGaussianNoise 这个属性。这意味着在你当前版本的 albumentations 库中可能没有这个特定的数据增强转换器。

为了解决这个问题，请尝试以下步骤：

1. 确认 albumentations 的安装版本。确保你的版本支持 IAAAdditiveGaussianNoise。你可以通过命令行运行以下代码来查看版本：

   pip show albumentations
   

   或者

   conda list albumentations
   

2. 如果当前版本不支持 IAAAdditiveGaussianNoise，你需要升级到一个更高版本。可以使用以下命令进行升级：

   pip install --upgrade albumentations
   

3. 升级完成后，重新运行你的代码片段。如果问题依然存在，那可能是安装过程中出现了问题或者该功能在新版库中被移除或重命名了。请查阅 albumentations 的官方文档（https://albumentations.ai/docs/api_reference/augmentations/transforms/）以确认最新API的用法。

若 IAAAdditiveGaussianNoise 功能确实已被移除，你可以考虑替换为类似的功能，例如仅使用 GaussNoise 或其他可用的噪声添加操作。以下是修改后的代码示例：

import albumentations as A
import cv2

p1 = 0.95
p2 = 0.85
p3 = 0.75
transform = A.Compose([
    A.RandomRotate90(p=p2),
    A.OneOf([
        A.GaussNoise(p=0.9),  # 替换了 IAAAdditiveGaussianNoise
        A.GaussNoise(p=0.6),
    ], p=p3)
], p=p1)

image = cv2.imread('some/image.jpg')
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
transformed = transform(image=image)
transformed_image = transformed['image']

这样应该能解决你的问题。如果仍有疑问，请进一步提供相关信息。