Clear Filters
Clear Filters

how to augment image data only for a specific class?

14 views (last 30 days)
I have a dataset of three class labeled as '0','1' and '3' stored in folder 'dataset'. The number of sample for the class '0' very less compare to other two class. If I use 'imageDataAugmenter' to augment data, It augments for all classes that doesn't solve my imblance problem.
Can anyone help me to advice any way that I can augment data for only class '0'.
Note: The three class data is in three differnt sub-folder within the main folder 'dataset'.

Answers (2)

Greg Heath
Greg Heath on 1 Jan 2020
Separate class 0 and interpolate. If you have a good feel for the data you could extrapolate. However the latter might be tricky.
Hope this helps.
Greg

Mahmoud Afifi
Mahmoud Afifi on 7 Jan 2020
You can use a custom function to augment the data of this class before training. You can try this one if the problem is related to natural images: https://www.mathworks.com/matlabcentral/fileexchange/72966-wb-color-augmenter?s_tid=prof_contriblnk

Categories

Find more on Recognition, Object Detection, and Semantic Segmentation in Help Center and File Exchange

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!