albumentations报错 KeyError‘You have to pass data to augmentations as named arguments,...... ‘

最近初入语义分割的坑，用albumentations库进行数据增强的时候报错如下：

网上也没搜到合适的解决办法，记录一下谨防以后再遇到这个问题。

根据报错提示：“You have to pass data to augmentations as named arguments, for example: aug(image=image)”，意思就是需要需要传数据参数给augmentations中的image形参，于是修改为如下就可以了：

image = trfm(image=images)
image = image["image"]