e9d5373461b0a037716722a461cb12416aa5d1bc,evaluate.py,,,#,44

Before Change

    CMC_tmp = evaluate(query_feature[i],query_label[i],query_cam[i],gallery_feature,gallery_label,gallery_cam)
    if CMC_tmp[0]==-1:
        continue
    CMC = torch.cat((CMC,CMC_tmp.view(1,len(CMC_tmp))), 0)
    print(i, CMC_tmp[0])

CMC = CMC.float()
CMC = CMC.mean(dim=0) //average CMC

After Change

gallery_cam = result["gallery_cam"][0]
gallery_label = result["gallery_label"][0]

CMC = torch.IntTensor(len(gallery_label)).zero_()
ap = 0.0
//print(query_label)
for i in range(len(query_label)):
    ap_tmp, CMC_tmp = evaluate(query_feature[i],query_label[i],query_cam[i],gallery_feature,gallery_label,gallery_cam)
    if CMC_tmp[0]==-1:
        continue
    CMC = CMC + CMC_tmp
    ap += ap_tmp
    print(i, CMC_tmp[0])

CMC = CMC.float()

Instances