albumentations-team / autoalbument

AutoML for image augmentation. AutoAlbument uses the Faster AutoAugment algorithm to find optimal augmentation policies. Documentation - https://albumentations.ai/docs/autoalbument/

Home Page:https://albumentations.ai/docs/autoalbument/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Shape of 3D affine matrices

varunj opened this issue · comments

I am getting the following error. I am using a custom segmentation network, but not sure if that is what is causing the error. Seems like instead of b number of 3x4 matrices the program is producing a single (2x2)x3 matrix.

image