How to check number of augmented images from imageDataAugmenter
8 次查看(过去 30 天)
显示 更早的评论
Hi, I am new to matlab and wanted to ask if there is a way to know the number of output images after augmentation from imageDataAugmenter.
As I have a high number of imbalanced classes, I wanted to know the class distribution after augmentation. Thanks in advance! This is how I augment the images:
imageAugmenter = imageDataAugmenter('RandRotation',[-180 180],'RandXReflection', true, 'RandYReflection', true, 'RandXShear', [-90, 90], 'RandYShear', [-90, 90]);
imageSize = [50 50 3];
datasource = augmentedImageSource(imageSize,imds70,'DataAugmentation',imageAugmenter);
0 个评论
回答(2 个)
Mahmoud Afifi
2020-5-7
ImageDataAugmenter does the augmentation during image loading, which means it doesn’t rebalance your data. It applies the selected augmentation on the fly.
If you would like to rebalance your data, you need to apply the augmentation offline and adjust the number of augmented samples accordingly. You can design your own custom function to do that. I refer you to our WB augmenter here .
Another option is to use a weighted loss func that penalizes more for mistakes in the classes that have lower training examples.
0 个评论
Kenta
2020-7-14
Hi, you can rebalance the number of images per class using the following sample. Just using the image augmenter as you mentioned does not compensate the class imbalance. Hope this helps.
0 个评论
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!