mirrors/fast-reid
1
0
Fork 0
mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00
Code Issues Projects Releases Wiki Activity

Update dsr_distance.py

Browse Source
This commit is contained in:
Lingxiao He 2020-07-12 13:31:21 +08:00 committed by GitHub
parent f8d468647c
commit 115de22532
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
projects/PartialReID/partialreid/dsr_distance.py
View File

@ -39,5 +39,6 @@ def compute_dsr_dist(array1, array2, distmat, scores, topk=30):
dist[i, index[i, j]] = (min_value * score).sum()
dist = dist.cpu()
dist = dist.numpy()
dist = 0.98 * dist + 0.02 * distmat
return dist