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 Evaluation

Browse Source
This commit is contained in:
Lingxiao He 2020-06-10 18:16:58 +08:00 committed by GitHub
parent 7f753027c5
commit 896933bf5e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
projects/PartialReID/partialreid/dsr_evaluation.py
View File

@ -65,19 +65,9 @@ class DsrEvaluator(DatasetEvaluator):
logger.info("Testing without DSR setting") logger.info("Testing without DSR setting")
self._results = OrderedDict() self._results = OrderedDict()
if self.cfg.TEST.DSR.ENABLED: if self.cfg.TEST.DSR.ENABLED:
max_value = 0
fusion_weight = 0
topk = self.cfg.TEST.DSR.TOPK topk = self.cfg.TEST.DSR.TOPK
dsr_dist = compute_dsr_dist(spatial_features[:self._num_query], spatial_features[self._num_query:], dist, dist = compute_dsr_dist(spatial_features[:self._num_query], spatial_features[self._num_query:], dist,
scores[:self._num_query], topk) scores[:self._num_query], topk)
for i in range(0, 101):
lamb = 0.01 * i
fusion_dist = (1 - lamb) * dist + lamb * dsr_dist
cmc, all_AP, all_INP = evaluate_rank(fusion_dist, query_pids, gallery_pids, query_camids, gallery_camids)
if (cmc[0]>max_value):
fusion_weight = lamb
max_value = cmc[0]
dist = (1 - fusion_weight) * dist + fusion_weight * dsr_dist
logger.info("Testing with DSR setting") logger.info("Testing with DSR setting")
cmc, all_AP, all_INP = evaluate_rank(dist, query_pids, gallery_pids, query_camids, gallery_camids) cmc, all_AP, all_INP = evaluate_rank(dist, query_pids, gallery_pids, query_camids, gallery_camids)