From 43b42a651c647e1dd65a9f024410cbaeb006421d Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 3 Jul 2020 10:45:54 +0800 Subject: [PATCH] fix(dnn/cuda): fix indexing logic in psroi_pooling a variable relating to indexing was not computed correctly GitOrigin-RevId: 548c8f3f145c6d48212894fd6040331aff78a0b0 --- dnn/src/cuda/deformable_ps_roi_pooling/opr_impl.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dnn/src/cuda/deformable_ps_roi_pooling/opr_impl.cpp b/dnn/src/cuda/deformable_ps_roi_pooling/opr_impl.cpp index 526ebaa62..ea60e4081 100644 --- a/dnn/src/cuda/deformable_ps_roi_pooling/opr_impl.cpp +++ b/dnn/src/cuda/deformable_ps_roi_pooling/opr_impl.cpp @@ -32,7 +32,7 @@ void create_param(const DeformablePSROIPoolingBase* opr, p.sample_per_part = param.sample_per_part; p.trans_std = param.trans_std; p.scale = param.spatial_scale; - p.nr_cls = p.no_trans ? 1 : trans[0]; + p.nr_cls = p.no_trans ? 1 : trans[1] / 2; p.nr_bbox = rois[0]; p.IC = data[1]; p.IH = data[2]; -- GitLab