PaddleDetection icon indicating copy to clipboard operation
PaddleDetection copied to clipboard

Published 20 hours ago • verified publisher icon PaddlePaddle

请教关于mask rcnn的代码问题——_C_ops.generate_proposals_v2

Open DreamMaker777 opened this issue 11 months ago • 4 comments

问题确认 Search before asking

  • [X] 我已经搜索过问题，但是没有找到解答。I have searched the question and found no related answer.

请提出你的问题 Please ask your question

请教一个mask rcnn中的接口函数：_C_ops.generate_proposals_v2，搜了一下文档，并没有找到相关说明。此函数是否能够返回保留ROI对应所有输入的index？

DreamMaker777 avatar Mar 20 '24 06:03 DreamMaker777

我在paddle源文件 paddle/test/legacy_test/test_generate_proposals_v2_op.py 中找到了python的复现 generate_proposals_v2_in_python，请问两者的结果一样吗？只有速度上的区别吗？

DreamMaker777 avatar Mar 20 '24 07:03 DreamMaker777

嗯理论上结果应该一样的

lyuwenyu avatar Mar 21 '24 06:03 lyuwenyu

@lyuwenyu 您好，python中的generate_proposals_v2_in_python速度过慢，nms阶段有个for循环，我用paddle.vision.ops.nms替换，但是相较_C_ops.generate_proposals_v2还是慢了不少；同时，替换之后会在RPN的bs_rois_num_collect = paddle.concat(bs_rois_num_collect)处报错ValueError: (InvalidArgument) The axis is expected to be in range of [0, 0), but got 0，print此处向量为bs rois num collect [Tensor(shape=[],dtype=int32, place=Place(cpu), stop_gradient=True, 1000)。请问_C_ops.generate_proposals_v2能否输出保留ROI对应所有输入的index？或者如何修改此函数，我没有找到相关接口。多谢！

DreamMaker777 avatar Mar 21 '24 07:03 DreamMaker777

参考 https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.7/ppdet/modeling/ops.py#L1049

lyuwenyu avatar Mar 22 '24 07:03 lyuwenyu