search

hyz-xmaster / VarifocalNet

VarifocalNet: An IoU-aware Dense Object Detector
Apache License 2.0
346 stars 52 forks source link

Find matching configuration file for a model #2

Open Engin1980 opened 3 years ago

Engin1980 commented 3 years ago

Dear authors, thank you for your work and for your repo.

Is there a simple way how to find matching configuration .py file (from /configs folder) for a particular .pth file?

E.g., I am looking for the configuration files for models: vfnet_r50_dcn_ms_2x_47.8.pth vfnet_r2_101_dcn_ms_2x_51.1.pth

For ResNet 50 I have tried vrefnet_r50_fpn_1x_coco.py and vrefnet_r50_fpn_mstrain_2x_coco.py giving me mAP 0.91. For R2 I have tried vfnet_r2_101_fpn_mstrain_2x_coco.py, got error 'pth file is not valid checkpoint file'.

Thank you for your answer in advance.

Best regards MV

hyz-xmaster commented 3 years ago

Hi @Engin1980, thanks for interesting in our work.

vfnet_r50_dcn_ms_2x_47.8.pth -> vfnet_r50_fpn_mdconv_c3-c5_mstrain_2x_coco.py

vfnet_r2_101_dcn_ms_2x_51.1.pth -> vfnet_r2_101_fpn_mdconv_c3-c5_mstrain_2x_coco.py

The rule is : dcn -> mdconv, ms -> mstrain.

By the way, if you use vfnet_r2_101_fpn_mdconv_c3-c5_mstrain_2x_coco.py and corresponding model file in inference, please comment delattr(self, 'conv2') in res2net.py.