# Load the hdf5 files
from keras.models import load_model
resnet50 = keras.models.load_model('/content/drive/MyDrive/HDF5 Files/RestNet 50 best_model.hdf5')
resnet152 = keras.models.load_model('/content/drive/MyDrive/HDF5 Files/best_model_4.hdf5')
# Get the predictions from each model
predictions1 = resnet50.predict(images)
predictions3 = resnet152.predict(images)
# Combine the predictions using a majority vote
predictions = np.array([predictions1, predictions3])
predictions = np.mean(predictions, axis=0)
print(predictions)
。我得到的输出是9.9993783e-01 1.3912816e-06 6.0800008e-05 2.9077312e-09。这意味着什么?
1条答案
按热度按时间k3fezbri1#
从你的问题来看,你通过了多少张图片,以及你的问题中有多少个分类类别,这并不清楚。我将试着在一般意义上回答。
假设你将一批4个图像传递给模型
restnet50
和resnet152
,并且它们都在5个类别上训练,则每个模型将给予具有形状(4,5)的预测。要将它们组合起来以获得多数票,您应该编写如下代码:
最后的class_index给出了4张图片中每一张图片所属的类或类别的索引。我希望这能有所帮助。
更好的方式您可以使用更好的方式在tensorflow 中创建集合模型,如下所示: