e9d5373461b0a037716722a461cb12416aa5d1bc,evaluate.py,,evaluate,#Any#Any#Any#Any#Any#Any#,7

Before Change

    score = gf*query
    score = score.sum(1)
    // predict index
    s, index = score.sort(dim=0, descending=True)
    index = index[0:2000]
    //print(index[0:5])
    // good index

After Change

    //index = index[0:2000]
    // good index
    query_index = np.argwhere(gl==ql)
    camera_index = np.argwhere(gc==qc)

    good_index = np.setdiff1d(query_index, camera_index, assume_unique=True)
    junk_index1 = np.argwhere(gl==-2)
    junk_index2 = np.intersect1d(query_index, camera_index)
    junk_index = np.append(junk_index2, junk_index1) //.flatten())
    
    CMC_tmp = compute_mAP(index, good_index, junk_index)
    return CMC_tmp

Instances