最近初入语义分割的坑,用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"]

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐