search

tusen-ai / simpledet

A Simple and Versatile Framework for Object Detection and Instance Recognition
Apache License 2.0
3.08k stars 486 forks source link

in class RpnParam: num_reg_class = 81 why? #302

Closed dongzhenguo2016 closed 4 years ago

dongzhenguo2016 commented 4 years ago

在cascade_r101v1_fpn_1x.py中 class RpnParam: class bbox_target: num_reg_class = 2 # RPN只判断proposal是目标还是背景。 但是在faster_r101v1_fpn_1x.py中 class RpnParam: class bbox_target: num_reg_class = 81 # 为什么是81呢？我觉得应该是2才行呀？为何是coco数据集的类别数+1？ image thanks everyone! happy new year!

xchani commented 4 years ago

Both is fine. The only difference is class aware and class agnostic proposal.

huangzehao commented 4 years ago

The parameters in RpnParam.bbox_target are used for generating regression bbox targets in rcnn stage. And there are class aware and class agnostic setting in rcnn stage.