이미지 분류 ImageDataGenerator, flow_from_directory에서 class

Question

from tensorflow.keras.preprocessing.image import ImageDataGenerator train_datagen &#61; ImageDataGenerator( rescale &#61; 1.0/255. ) test_datagen &#61; ImageDataGenerator( rescale &#61; 1.0/255. ) train_generator &#61; train_datagen.flow_from_directory(train_dir, batch_size&#61;64, class_mode&#61;'binary', target_size&#61;(150, 150)) validation_generator &#61; test_datagen.flow_from_directory(validation_dir, batch_size&#61;64, class_mode &#61; 'binary', target_size &#61; (150, 150)) https://codetorial.net/tensorflow/classifying_the_cats_and_dogs.html 이 블로그를 참고해서 이미지 분류 연습을 하고 있는데 제가 만든 데이터셋으로 하고 있습니다.dataset 폴더 안에 train, validation 폴더를 만든 후 각각 동일하게 tree, flower 폴더를 만들었는데 저 코드를 실행시키면 train에 대해서는 class가 2개라고 나오는데 test는 class가 3개라고 출력됩니다.왜 test의 class는 3개라고 뜨는 지 모르겠습니다. Found 1250 images belonging to 2 classes. Found 550 images belonging to 3 classes. 코드텍스트

이미지 분류 ImageDataGenerator, flow_from_directory에서 class

0개의 답변

#cat 목록