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

Browse Source
This commit is contained in:
weishengyu 2021-06-05 16:02:22 +08:00
parent 2a72c57e92
commit cd7f606a5f
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

@ -353,12 +353,12 @@ class Trainer(object):
block_fea, gallery_feas, transpose_y=True)
if query_camera_id is not None:
camera_id_block = camera_id_blocks[block_idx]
camera_id_same = (camera_id_block == gallery_camera_id.t())
camera_id_same = (camera_id_block != gallery_camera_id.t())
image_id_block = image_id_blocks[block_idx]
image_id_same = (image_id_block == gallery_img_id.t())
image_id_same = (image_id_block != gallery_img_id.t())
keep_mask = not (camera_id_same and image_id_same)
keep_mask = paddle.logical_or(camera_id_same, image_id_same)
similarity_matrix = similarity_matrix * keep_mask.astype(
"float32")
if cum_similarity_matrix is None: