提交 81c864b9 编写于 作者: W weishengyu

dbg

上级 cd7f606a
...@@ -353,12 +353,12 @@ class Trainer(object): ...@@ -353,12 +353,12 @@ 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_same = (camera_id_block != gallery_camera_id.t()) camera_id_mask = (camera_id_block != gallery_camera_id.t())
image_id_block = image_id_blocks[block_idx] image_id_block = image_id_blocks[block_idx]
image_id_same = (image_id_block != gallery_img_id.t()) image_id_mask = (image_id_block != gallery_img_id.t())
keep_mask = paddle.logical_or(camera_id_same, image_id_same) keep_mask = paddle.logical_or(camera_id_mask, image_id_mask)
similarity_matrix = similarity_matrix * keep_mask.astype( similarity_matrix = similarity_matrix * keep_mask.astype(
"float32") "float32")
if cum_similarity_matrix is None: if cum_similarity_matrix is None:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册