shinmura0 / deepanomalydetection_benchmark Goto Github PK

    """
    テスト用
    model: 特徴抽出用モデル(predict)
    """
    model = Model(model.get_layer(index=0).input, model.get_layer(index=-4).output) # Arcfaceを削除
    
    # 正常vectorの呼び出し
    hold_vector = model.predict(train)# shape(len(train), 1280)

    # test_vectorの呼び出し
    predict_vector = model.predict(test)# shape(len(test), 1280)

    score = []

    for i in range(len(predict_vector)):
        cos_similarity = cosine_similarity(predict_vector[i], hold_vector) # shape(len(test), len(train))
        score.append(np.max(cos_similarity))

    return np.array(score)