Browse Source

fix(dnn/cuda): fix indexing logic in psroi_pooling

a variable relating to indexing was not computed correctly

GitOrigin-RevId: 548c8f3f14
release-0.5
Megvii Engine Team Xu Xinran 5 years ago
parent
commit
43b42a651c
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      dnn/src/cuda/deformable_ps_roi_pooling/opr_impl.cpp

+ 1
- 1
dnn/src/cuda/deformable_ps_roi_pooling/opr_impl.cpp View File

@@ -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];


Loading…
Cancel
Save