Làm thế nào để có được các nhãn id trong Keras khi đào tạo trên nhiều lớp học?

Question

Tôi đang sử dụng flow_from_directory để có được đào tạo thiết lập từ một thư mục với cấu trúc sau:

train 
    class1 
    class2 
    class3 
    ... 

Máy phát điện được gọi là nó sau:

train_generator = train_datagen.flow_from_directory( 
     train_data_dir,        
     target_size=(img_height, img_width),   
     batch_size=32,        
     class_mode='categorical') 

Tôi không thiết lập các tham số classes, nhưng tôi đã mong đợi để có được các nhãn theo thứ tự bảng chữ cái.

classes: optional list of class subdirectories (e.g. ['dogs', 'cats']). Default: None. If not provided, the list of classes will be automatically inferred (and the order of the classes, which will map to the label indices, will be alphanumeric).

Tuy nhiên, khi tôi phân loại hình ảnh đào tạo (để kiểm tra nhãn nào được trả lại), tôi không nhận được bất kỳ thứ tự cụ thể nào. Việc đào tạo diễn ra tốt (độ chính xác của ~ 85%), và có một sự nhất quán với các nhãn đầu ra khi phân loại các hình ảnh từ cùng một lớp.

Tôi làm cách nào để suy ra số nhãn được tạo bởi flow_from_directory và ánh xạ chúng tới các lớp?

Answer 1

Bạn có thể thấy lớp tương ứng với đó Integer nhìn vào biến ImageDataGenerator.class_indices

Dưới đây là một ví dụ về cách sử dụng nó

def build(source=None): 
     datagen = ImageDataGenerator(rescale=1./255) 
     data_generator = datagen.flow_from_directory(
     source, # this is the target directory 
     target_size=(150, 150), # all images will be resized to 150x150 
     batch_size=11, 
     class_mode='sparse') 
     class_dictionary = data_generator.class_indices 
    return data_generator, class_dictionary