hyz-xmaster/VarifocalNet

Find matching configuration file for a model

Engin1980 opened this issue · 1 comments

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

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.