我只用很少的数据训练了一个 epoch 的 CNN 模型。我使用 Keras 2.05。
这是 CNN 模型的(部分)最后 2 层number_outputs = 201。训练数据输出是一种热编码 201 输出。
model.add(Dense(200, activation='relu', name='full_2'))
model.add(Dense(40, activation='relu', name='full_3'))
model.add(Dense(number_outputs, activation='softmax'))
model.compile(loss='categorical_crossentropy', optimizer=adam, metrics=['accuracy'])
Run Code Online (Sandbox Code Playgroud)
模型保存为 h5 文件。然后,保存的模式将加载与上面相同的模型。batch_image是一个图像文件。
prediction = loaded_model.predict(batch_image, batch_size=1)
Run Code Online (Sandbox Code Playgroud)
我得到这样的预测:
ndarray: [[ 0.00498065 0.00497852 0.00498095 0.00496987 0.00497506 0.00496112
0.00497585 0.00496474 0.00496769 0.0049708 0.00497027 0.00496049
0.00496767 0.00498348 0.00497927 0.00497842 0.00497095 0.00496493
0.00498282 0.00497441 0.00497477 0.00498019 0.00497417 0.00497654
0.00498381 0.00497481 0.00497533 0.00497961 0.00498793 0.00496556
0.0049665 0.00498809 0.00498689 0.00497886 0.00498933 0.00498056
Run Code Online (Sandbox Code Playgroud)
问题:
预测数组应该是1, …