Не удалось загрузить пользовательский набор данных

Question

Я хочу обучить классификатор CNN в керасе, но мой набор данных сделан на заказ.Я создал модель CNN, но я совершенно не понимаю, как ввести свой набор данных в наборы поездов, действительные и тестовые наборы.Я новичок в углубленном изучении.У меня есть 7 типов классов, а именно ['A','C','F','L','N','R','V'], но они включены в названия изображений.Мои имена изображений будут как MLII_C_1.jpeg or V1_C_1.jpeg.Средняя буква в именах изображений дает мне класс, к которому он принадлежит.Как я могу загрузить мои изображения?и сделать по целевым меткам.Пожалуйста, кто-нибудь, помогите мне.

Answer 1

Я считаю, ImageDataGenerator () - это один из самых простых способов сделать это. Предположим, у вас есть данные (изображения), разделенные на обучение, проверку и тестирование в следующей иерархии:

train-| class1
      | class2 
      . 
      .
      | classN 


valid-| class1
      | class2 
      . 
      .
      | classN 


test- | class1
      | class2 
      . 
      .
      | classN 

затем в python начните с указания пути как:

from keras.preprocessing.image import ImageDataGenerator

train_path = "Path"
valid_path = "Path"
test_path =  "Path"

, затем просто используйте это:

trainBatches = ImageDataGenerator().flow_from_directory(train_path, target_size=(224,224), classes=['class1', 'class2', .. , 'classn'], batch_size=64)
valBatches = ImageDataGenerator().flow_from_directory(valid_path, target_size=(224,224), classes=['class1', 'class2', .. , 'classn'], batch_size=32)
testBatches = ImageDataGenerator().flow_from_directory(test_path, target_size=(224,224), classes=['class1', 'class2', .. , 'classn'], batch_size=32)