search

liming-ai / AlignDet

Official code for ICCV 2023 Paper: AlignDet: Aligning Pre-training and Fine-tuning in Object Detection.
https://liming-ai.github.io/AlignDet/
Apache License 2.0
142 stars 13 forks source link

pretrained weights for Faster-RCNN #30

Open zcablii opened 1 month ago

zcablii commented 1 month ago

Thank you for your excellent work! It seems that the link provided in the README for the pretrained Faster-RCNN weights is not correct as it links to mask-rcnn weights. Could you kindly update the links or provide an alternative way to access the weights?

liming-ai commented 1 month ago

Thank you for your excellent work! It seems that the link provided in the README for the pretrained Faster-RCNN weights is not correct as it links to mask-rcnn weights. Could you kindly update the links or provide an alternative way to access the weights?

Hi @zcablii. Sorry for the late reply. The Faster R-CNN and the Mask R-CNN has the same pretrained weights since we only pretrain the detection head. During fine-tuning stage, Mask R-CNN's mask head is initialized randomly and the detection head is pretrained. Please note here the only difference between Mask R-CNN and Faster R-CNN is whether has a mask head