RT-DETR中的where - Githubissues

PaddlePaddle / PaddleDetection

Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.

Apache License 2.0

12.86k stars 2.9k forks source link

RT-DETR中的where #8548

Open sdreamforchen opened 1 year ago

sdreamforchen commented 1 year ago

问题确认 Search before asking

[X] 我已经查询历史issue，没有发现相似的bug。I have searched the issues and found no similar bug report.

Bug组件 Bug Component

No response

Bug描述 Describe the Bug

https://github.com/PaddlePaddle/PaddleDetection/blob/a3e9c925074dd91f98516c2bac057d190af9d7aa/ppdet/modeling/transformers/rtdetr_transformer.py#L508 以及https://github.com/PaddlePaddle/PaddleDetection/blob/a3e9c925074dd91f98516c2bac057d190af9d7aa/ppdet/modeling/transformers/rtdetr_transformer.py#L524C16-L524C16 您好，我自己单独把这个_generate_anchors写了一个python跑了一下，valid_mask恒为true： 1 onnx导出为false 2 这两个where是否有必要哇。应该是我理解错了。麻烦解答一下，谢谢！！！

复现环境 Environment

develop分支

Bug描述确认 Bug description confirmation

[X] 我确认已经提供了Bug复现步骤、代码改动说明、以及环境信息，确认问题是可以复现的。I confirm that the bug replication steps, code change instructions, and environment information have been provided, and the problem can be reproduced.

是否愿意提交PR？ Are you willing to submit a PR?

[X] 我愿意提交PR！I'd like to help by submitting a PR!

lyuwenyu commented 1 year ago

其实他是为了保证生成的anchors是个合格的bbox 在看前面生成的时候有些超参数控制的不排除在某些条件下生成的box有问题所以才有这个逻辑 https://github.com/PaddlePaddle/PaddleDetection/blob/a3e9c925074dd91f98516c2bac057d190af9d7aa/ppdet/modeling/transformers/rtdetr_transformer.py#L505-L506