coincheung / triplet-reid-pytorch Goto Github PK

Contributors

Stargazers

Forkers

804463592 deep0learning upgirlnana yanbai1993 p1uto0 daehyun-bae feesics zj-dong jarygrace nicksonyap sanduanji choodly gzshan ljl02521 theoseo daiguangzhao back2yes ronybenitez buzyeli linhandai ives-kwy abandonsea leihao1999 yashenkoxciv alpsencer68

for it, (img, lb_id) in enumerate(dl):
    print('\r=======>  processing iter {} / {}'.format(it, all_iter_nums),
            end = '', flush = True)
    label_ids.append(lb_id)
    #label_cams.append(lb_cam)
    embds = []
    for im in img:
        im = im.cuda()
        embd = model(im).detach().cpu().numpy() #通过网络生成特征向量
        embds.append(embd)
    embed = sum(embds) / len(embds)
    embeddings.append(embed)

mAP is: 0.525827546826, cmc is: [0.7185273]