python - 将 AdditiveGaussianNoise 添加到单个图像 - AssertionError: Expected boolean as argument for 'return_batch'
问题描述
我想将AdditiveGaussianNoise
(链接:https ://imgaug.readthedocs.io/en/latest/source/overview/arithmetic.html#additivegaussiannoise )添加到我之前调整大小的单个图像中。
这是我的代码:
from skimage.io import imread
from skimage.transform import resize
import imgaug.augmenters as iaa
file_name = "path/to/image.jpg"
resized_img = resize(imread(file_name), (224, 224))
aug = iaa.AdditiveGaussianNoise(scale=(0, 0.2*255))
augmented_image = aug(resized_img)
我收到此错误消息:
---------------------------------------------------------------------------
AssertionError Traceback (most recent call last)
<ipython-input-20-e4a0b17d4ac4> in <module>()
----> 1 augmented_image =aug(resized_img)
1 frames
/usr/local/lib/python3.6/dist-packages/imgaug/augmenters/meta.py in augment(self, return_batch, hooks, **kwargs)
1782 ("Expected boolean as argument for 'return_batch', got type %s. "
1783 + "Call augment() only with named arguments, e.g. "
-> 1784 + "augment(images=<array>).") % (str(type(return_batch)),)
1785 )
1786
AssertionError: Expected boolean as argument for 'return_batch', got type <class 'numpy.ndarray'>. Call augment() only with named arguments, e.g. augment(images=<array>).
我该如何修改我的代码?
非常感谢!
解决方案
这是使用随机图像的解决方案...您需要指定images
参数
import numpy as np
import imgaug.augmenters as iaa
img = np.random.randint(0,256, (1,224,224,3)).astype('float32')
aug = iaa.AdditiveGaussianNoise(scale=(0, 0.2*255))
augmented_image = aug(images=img)
推荐阅读
- android - 在 Activity 上下文之外调用 startActivity() 需要 FLAG_ACTIVITY_NEW_TASK
- docker - 使用卷挂载文件
- c - x86系统调用表中的交换函数
- c++ - 自动生成的顶点缓冲区和索引缓冲区不起作用
- javascript - div 背景和边框颜色未显示
- json - 如何在 swift 4.2 中将 JSON 转换为 Codable?
- node.js - 如何在使用 webpack 热模块重新加载启动新实例之前等待应用程序关闭?
- android - 如何使用 ExoPlayer 提取一段音频?
- flutter - (背景)图像未达到屏幕宽度
- c# - ARKit Unity 从场景中删除预制件