![]() I am trying to do the following: datagen = ImageDataGenerator(horizontal_flip=True, vertical_flip=True, rescale=1./255, class_mode="multi-label") ![]() ![]() But, I am not able to apply data augmentation. I am able to train the network without Keras data augmentation (for two outputs). However, I am not able to figure out how I can pass two label arrays because, it is taking the labels as (2, number of samples) when I send it as list. So, accordingly, I have formatted my data to have X_train as a (number of samples, height, width, channels) numpy array, y_train1 as (number of samples,) numpy array and y_train2 as (number of samples,) numpy array.Īs I am not training using directory structure, I am using ImageDataGenerator.flow(). ![]() I have a transfer learning based two output classification problem. ![]()
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. Archives
March 2023
Categories |