विरल और स्पष्ट कक्षाओं के लिए Keras ImageDataGenerator प्रवाह वर्ग के सूचकांक

वोट
0

Keras के लिए ImageDataGenerator flow_*विधियों, यह स्पष्ट और विरल के लिए वर्ग सूचकांकों की एक stringified संस्करण की आवश्यकता है class_mode। मैं कक्षा लेबल उस तरह देखो ['0','1',...,'10','11',...]और एक दुर्भाग्यपूर्ण परिणाम यह है कि स्ट्रिंग-वर्णमाला के क्रम में Keras अनुक्रमित इन:

उदाहरण के लिए:

datagen = ImageDataGenerator(
        rotation_range=0,
        width_shift_range=0,
        height_shift_range=0,
        rescale=None,
        shear_range=0,
        zoom_range=0,
        horizontal_flip=False,
        preprocessing_function=preprocessor,
        fill_mode='nearest')

test_generator = datagen.flow_from_dataframe(
    dataframe=dfTest,
    directory=None,
    x_col=filePath,
    y_col=ycat,
    target_size=SIZE,
    batch_size=BATCH_SIZE,
    class_mode='sparse',
    shuffle=False)

    print(test_generator.class_indices)

देता है:

{'0': 0,
 '1': 1,
 '10': 2,
 '11': 3,
 ...,
 '2': 12,
 '20': 13,
 '21': 14,
 '22': 15,
 '3': 16,
 '4': 17,
 '5': 18,
 '6': 19,
 '7': 20,
 '8': 21,
 '9': 22}

आदर्श रूप में मैं देखना चाहते हैं:

{'0': 0,
 '1': 1,
 '2': 2,

 ...,
 }

मैंने सोचा था कि के बारे में मैन्युअल रूप से बदल रहा है test_generator.class_indices, लेकिन मुझे यकीन है कि के रूप में प्रारंभ के बाद करने के लिए सुरक्षित है नहीं कर रहा हूँ, जनरेटर पूर्व की गणना की है डेटासेट के वर्ग लेबल।

वहाँ फिर से लिखने के बिना इस के लिए एक अच्छा समाधान है flow_*विधि?

14/02/2020 को 00:02
का स्रोत उपयोगकर्ता
अन्य भाषाओं में...                            

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more