distmat = torch.pow(query_feats, 2).sum(dim=1, keepdim=True).expand(m, n) + torch.pow(gallery_feats, 2).sum(dim=1, keepdim=True).expand(n, m).t() - Githubissues

songwsx / person_search_demo

利用YOLOv3结合行人重识别模型，实现行人的检测识别，查找特定行人

MIT License

531 stars 135 forks source link

distmat = torch.pow(query_feats, 2).sum(dim=1, keepdim=True).expand(m, n) + torch.pow(gallery_feats, 2).sum(dim=1, keepdim=True).expand(n, m).t() #16

Open mojiayuan opened 4 years ago

mojiayuan commented 4 years ago

您好，请问下这个distmat.addmm_(1, -2, query_feats, gallery_feats.t()) 和上面的是哪个距离的公式的