search

jshilong / GroupRCNN

Group R-CNN for Point-based Weakly Semi-supervised Object Detection (CVPR2022)
Apache License 2.0
139 stars 10 forks source link

The code #1

Closed jianpingZhonggit closed 2 years ago

jianpingZhonggit commented 2 years ago

Nice work! However， the code seems not complete.​ For example, in the paper, you use retinaHead to replace RPNHead, but, you do not offer the fixed retinaHead. Hope you can update those incomplete code. Thanks!​

jshilong commented 2 years ago

The code support adopts any one-stage detector as RPN by modifying the config. https://github.com/jshilong/GroupRCNN/blob/31ec3eb32c0c1004138a4457b52a4ce822d99248/projects/configs/10_coco/group_rcnn_24e_10_percent_coco_detr_augmentation.py#L20

So there is no any modification for RetinaHead, we just import it from mmdetection, you can launch this project after install MMCV& MMDetection follow our readme.

jshilong commented 2 years ago

Feel free to reopen the issue if there is any question.