sir after all the loading i trained the model the no of images in train dataset is 364 only. so i want to keep all images corresponding to different classes equally in my train dataset and validation dataset . how can i do that? i am currently using mobilenet_v2 as model since images are less should i write my own model or using this model what  can i do to increase accuracy?