【狗狗分类项目】(4)小功能编写

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"

猜你喜欢

转载自blog.csdn.net/weixin_41770169/article/details/80531701