search

open-mmlab / mmdetection

OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io
Apache License 2.0
29.43k stars 9.43k forks source link

How to download res2net 50 pre-trained model #3700

Closed minan19605 closed 4 years ago

minan19605 commented 4 years ago

How to download res2net 50 pre-trained model? There only have res2net 101 model.

Thanks a lot!

xvjiarui commented 4 years ago

Hi @minan19605 You may refer to the official implementation here. Note that you may need to convert some miss matching keys.

shinya7y commented 4 years ago

Please try the below. model key conversion script

minan19605 commented 4 years ago

@shinya7y Thank you so much. the model works very well

minan19605 commented 4 years ago

I download the pth file into my PC and scp to the /home/xxxx/.cache/torch/checkpoints. Then modify the config py file as below:

model = dict( pretrained= '/home/xxx/.cache/torch/checkpoints/res2net50_v1b_26w_4s-3cf99910_mmdetv2pth', backbone=dict(type='Res2Net', depth=50, scales=4, base_width=26))