轻松掌握 MMDetection 中 Head 流程

庄守义 · 发表于 2023-1-12 13:47:35

请问loss /loss_single函数是怎么传参的？

千邱 · 发表于 2023-1-12 13:48:12

不好意思打扰了。“其中 forward 方法用于运行 head 网络部分输出分类回归分支的特征图”，之前的文章里也有类似的话，为什么forward 方法是输出特征图呢？不应该是分类与回归分支的结果吗，也就是回归得到的偏置参数（offset）和预测的类别。请赐教。

圣尚 · 发表于 2023-1-12 13:48:28

bbox head在推理时候实际上分成了两个函数，forward仅仅是推理特征图，get_bboxes函数会基于该特征图得到最终的分类和回归直接

周晓波 · 发表于 2023-1-12 13:48:38

很棒！！

把自己弄丢了 · 发表于 2023-1-12 13:49:12

第一章‘1 Head 模块整体概述’，前四段。
这里的denseHead 和ROI head 是不是说的有点混乱和错误。
（1）我记得ROIHead通常用于region proposal 以及前景背景的二分类；ROI head对应二阶段检测器中的第一阶段。
（2）denseHead用于多分类，box 回归与微调，方向回归，mask预测等；这里对应二阶段检测器中的第二阶段，以及多数一阶段检测器。 @OpenMMLab

虾米无名 · 发表于 2023-1-12 13:49:32

请问在two-stage的训练阶段中rpn部分的dense_head不需要调用getboxes把proposal传递给roi_head吗,因为我看文章说尽在测试阶段才调用getboxes，所以有疑问

周建群 · 发表于 2023-1-12 13:50:07

不会的，他本质就是 for 循环

当代语文 · 发表于 2023-1-12 13:50:50

能不能讲解一下检测模型怎样部署，比如导出onnx、trt等

相先生 · 发表于 2023-1-12 13:51:37

部署问题我们现在发布了全新的 mmdeploy 框架，支持大部分 OpenMMLab 算法库的模型，欢迎试用。GitHub - open-mmlab/mmdeploy: OpenMMLab Model Deployment Framework

天艺广告 · 发表于 2023-1-12 13:51:44

先进是先进，就是本来只用会pytorch就差不多可以看懂代码，现在又需要学mmdet[大哭]

		自动登录	找回密码
密码			立即注册