xingyizhou / CenterNet2

Two-stage CenterNet
Apache License 2.0
1.2k stars 189 forks source link

Not found Base-CenterNet2.yaml. #29

Open sunset326 opened 3 years ago

sunset326 commented 3 years ago

Hi,i want to use CenterNet2_R2-101-DCN-BiFPN_4x+4x_1560_ST.yaml to demo ,and i have download CenterNet2_R2-101-DCN-BiFPN_4x+4x_1560_ST.yaml and CenterNet2_R2-101-DCN-BiFPN_4x+4x_1560_ST.pth, but i can't find the file of Base-CenterNet2.yaml.so how can i download it.thank you.

xingyizhou commented 3 years ago

Hi, you can found the config here.

sunset326 commented 3 years ago

Thank you for your rapid response,i have download the model,but i have a new error: raise KeyError("Non-existent config key: {}".format(full_key)) KeyError: 'Non-existent config key: MODEL.BIFPN'

xingyizhou commented 3 years ago

Hi, can you refer to this and see if the issue can be solved?

sunset326 commented 3 years ago

Sorry,i have the same file,but i can't solve the problem

GrantorShadow commented 3 years ago

Hey @sunset326, not sure if you solved the error but here's what I did, following the solution given in #30

Add this to imports

from projects.CenterNet2.centernet.config import add_centernet_config

Add this line to after get_cfg() command in the setup_cfg() function

add_centernet_config(cfg)

You just need to update the config keys with the ones used by centernet2