1 随机展示一张照片:先不展示
2 人脸加狗耳朵、狗鼻子、狗嘴巴
# 获取预测向量
predicted_vector = Resnet50InV3X_model.predict(bottleneck_feature)
#print(predicted_vector)
print(np.max(predicted_vector))
# 返回此模型预测的狗的品种
if np.max(predicted_vector) > 0.6:
return dog_names[np.argmax(predicted_vector)]
else:
sortLoc = np.argsort(predicted_vector)
sortVal = np.sort(predicted_vector)
print(dog_names[sortLoc[0]], 'has probability ', sortVal[0])
print(dog_names[sortLoc[1]], 'has probability ', sortVal[1])
print(dog_names[sortLoc[2]], 'has probability ', sortVal[2])
return "Mongrel"
3 杂交狗检测
# 获取预测向量
predicted_vector = Resnet50InV3X_model.predict(bottleneck_feature)
#print(predicted_vector)
print(np.max(predicted_vector))
# 返回此模型预测的狗的品种
if np.max(predicted_vector) > 0.6:
return dog_names[np.argmax(predicted_vector)]
else:
sortLoc = np.argsort(predicted_vector)
sortVal = np.sort(predicted_vector)
print(dog_names[sortLoc[0]], 'has probability ', sortVal[0])
print(dog_names[sortLoc[1]], 'has probability ', sortVal[1])
print(dog_names[sortLoc[2]], 'has probability ', sortVal[2])
return "Mongrel"