search

xingyizhou / CenterNet2

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

when I use custom dataset to train CenterNet2_R2-101-DCN-BiFPN_1280_4x ,the project is break. #11

Open ZJU-lishuang opened 3 years ago

ZJU-lishuang commented 3 years ago 
File "/home/lishuang/Disk/gitlab/traincode/CenterNet2/projects/CenterNet2/centernet/modeling/backbone/bifpn_fcos.py", line 240, in forward
    raise NotImplementedError()
NotImplementedError

h =1 target_h =1 w =2 target_w =1 How to deal with it?

Linchunhui commented 3 years ago

I have the same problem when i train CenterNet2_DLA-BiFPN-P5_640_24x_ST.yaml

xiaowanzizz commented 3 years ago

I also have the same problem when i train CenterNet2_DLA-BiFPN-P5_640_24x_ST.yaml hope to get some help

xingyizhou commented 3 years ago

Hi all, Thank you for bringing up the issue and sorry for the very delayed response. It is caused by incorrect size divisibility in the fcos_bifpn backbone. I have fixed this in the most recent commit and the error should have gone. Strangely I didn't meet this before when I train the model. This may indicate there are some unnoted differences between the release code and my development version. I will check this.

XYSC commented 3 years ago

When I train CenterNet2_DLA-BiFPN-P3_24x.yaml, I have a problem.

File "/data/anaconda3/envs/detectron2/lib/python3.8/site-packages/yacs/config.py", line 491, in _merge_a_into_b raise KeyError("Non-existent config key: {}".format(full_key)) KeyError: 'Non-existent config key: MODEL.ROI_BOX_HEAD.MULT_PROPOSAL_SCORE'

hosea7456 commented 3 years ago

When I train CenterNet2_DLA-BiFPN-P3_24x.yaml, I have a problem.

File "/data/anaconda3/envs/detectron2/lib/python3.8/site-packages/yacs/config.py", line 491, in _merge_a_into_b raise KeyError("Non-existent config key: {}".format(full_key)) KeyError: 'Non-existent config key: MODEL.ROI_BOX_HEAD.MULT_PROPOSAL_SCORE'

@XYSC @xingyizhou I also facing the same problem.

huyphan168 commented 2 years ago

You can use this below code to add new config keys: from detectron2.projects.CenterNet2.centernet.config import add_centernet_config cfg = get_cfg() add_centernet_config(cfg)