mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

dbg camera

Browse Source
This commit is contained in:
weishengyu 2021-06-05 15:35:20 +08:00
parent 19be351646
commit 5597830e4c
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ppcls/engine/trainer.py
View File

@ -351,9 +351,9 @@ class Trainer(object):
block_fea, gallery_feas, transpose_y=True) block_fea, gallery_feas, transpose_y=True)
if query_camera_id is not None: if query_camera_id is not None:
camera_id_block = camera_id_blocks[block_idx] camera_id_block = camera_id_blocks[block_idx]
camera_id_mask = (camera_id_block != gallery_camera_id) camera_id_mask = (camera_id_block != gallery_camera_id.t())
similarity_matrix = similarity_matrix.masked_select( similarity_matrix = similarity_matrix * camera_id_mask.astype(
camera_id_mask) "float32")
if cum_similarity_matrix is None: if cum_similarity_matrix is None:
cum_similarity_matrix = similarity_matrix cum_similarity_matrix = similarity_matrix
else: else: