提交 6078187e 编写于 作者: M Megvii Engine Team

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

a variable relating to indexing was not computed correctly

GitOrigin-RevId: 548c8f3f145c6d48212894fd6040331aff78a0b0
上级 32d91d5e
......@@ -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];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册