Keras-TF model.predict() 给了我错误的结果

Question

我刚刚训练了我的 CNN 模型，该模型具有以下层详细信息：

model = Sequential()
model.add(Conv2D(100, (3, 3), input_shape=(100, 100, 1), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(120, (5, 5), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Conv2D(140, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(rate=0.5))
model.add(Flatten())
model.add(Dense(200, activation='relu'))
model.add(Dropout(rate=0.5))
model.add(Dense(100, activation='relu'))
model.add(Dropout(rate=0.5))
model.add(Dense(num_classes, activation='softmax'))
model.compile(Adam(lr=0.001), loss='categorical_crossentropy', metrics=['accuracy'])

我的模型在 keras 验证/测试结果上的准确率约为 95.xxx%。

当我导入 h5 文件来测试图像的预测时，我做到了：

def grayscale(img):
  img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
  return img

def equalize(img):
  img = cv2.equalizeHist(img)
  return img

def preprocessing(img):
  gray = grayscale(img)
  eq = equalize(gray)
  return eq / 255

image = cv2.imread("sample.jpg")
print("Loading model...")
model = load_model("model.h5")
classes = pd.read_csv('dataset.csv', header=0, usecols=['BananaOrNot']).values

image = preprocessing(image)
image = image.reshape(100, 100, 1)
image = np.expand_dims(image, axis=0)
y_prob = model.predict(image)
class_idx = y_prob.argmax(axis=-1)[0]
print(classes[class_idx][0]) # it produced wrong result (should be 'not banana', got 'banana')

我用一个示例 空白 图像进行了测试，预测给了我 香蕉 结果，尽管它应该是 不是香蕉 .在这个非常简单的测试用例中怎么可能出错？还是我的model.predict(img) 输入图片有什么问题？

Answer 1

由于您的数据集中没有名为 Notbanana 的类，因此它仅提供您在网络中训练过的类。

为了在您的模型上具有该行为，您应该在模型上设置一个类例如：背景，并使用一定数量的背景标记图像对其进行训练。否则它会给出输出向量中最可能的结果即最可能的类，即使它是空白图像。

或

您可以训练分类模型以及本地化和分类仅发生在本地化对象中，您的问题可能会得到纠正。