From 5597830e4c0aa38e7ec2103b8bc90ded29ad933c Mon Sep 17 00:00:00 2001 From: weishengyu Date: Sat, 5 Jun 2021 15:35:20 +0800 Subject: [PATCH] dbg camera --- ppcls/engine/trainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ppcls/engine/trainer.py b/ppcls/engine/trainer.py index d599f6770..e37b49780 100644 --- a/ppcls/engine/trainer.py +++ b/ppcls/engine/trainer.py @@ -351,9 +351,9 @@ 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_mask = (camera_id_block != gallery_camera_id) - similarity_matrix = similarity_matrix.masked_select( - camera_id_mask) + camera_id_mask = (camera_id_block != gallery_camera_id.t()) + similarity_matrix = similarity_matrix * camera_id_mask.astype( + "float32") if cum_similarity_matrix is None: cum_similarity_matrix = similarity_matrix else: