mmcv里的nms、softnms代码实现片段在哪里？

采用pytorch1.6、mmcv1.15、mmdetection2.6版本的Faster R-CNN网络框架，想改进softnms。按照提示打开了mmcv里的nms.py文件，不知道softnms实现线性linear和gaussian的代码片段在哪里？

建议把完整代码发一下。如有帮助请采纳

您好，我是有问必答小助手，您的问题已经有小伙伴解答了，您看下是否解决，可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案，可以点击【采纳】按钮，给回答的小伙伴一些鼓励哦～～

ps:问答VIP仅需29元，即可享受5次/月 有问必答服务，了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632

你好，我也遇到了这个问题，请问mmcv里面他具体实现的代码在哪里啊？

1. cpu核心代码在ops/csrc/pytorch/cpu/nms.cpp
2. gpu核心代码包括两个部分，1）Tensor部分代码和cpu代码对应的gpu里面。2）kernel部分代码在common里面。

• 直接修改的话覆盖源代码进行编译即可
• 添加别名修改，需要进行注册、分发、绑定