MATLAB Answers

how to augment image data only for a specific class?

38 views (last 30 days)
Casio Uploader
Casio Uploader on 22 Dec 2019
Answered: Mahmoud Afifi on 7 Jan 2020
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'.

  0 Comments

Sign in to comment.

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

  0 Comments

Sign in to comment.


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

  0 Comments

Sign in to comment.

Community Treasure Hunt

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

Start Hunting!