Shape of 3D affine matrices

albumentations-team / autoalbument

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

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

MIT License

203 stars 20 forks source link

Shape of 3D affine matrices #20

Closed varunj closed 3 years ago

varunj commented 3 years ago

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.