2 classification task at the same time

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

198 stars 20 forks source link

2 classification task at the same time #37

Open iumyx2612 opened 2 years ago

iumyx2612 commented 2 years ago

Hello I'm building a multi task classification, my model contains 1 feature extractor and 2 classification heads, so when implementing __getitem()__ it needs to return 1 image with 2 labels Is this any way to implement auto augment that way?