search

Tung-I / Dual-awareness-Attention-for-Few-shot-Object-Detection

95 stars 16 forks source link

predefined backbone models #16

Open alphacyp opened 2 years ago

alphacyp commented 2 years ago

I wonder where can I download the predefined backbone models (e.g., res50, vgg16)? Can you give me a link？

Jessy-Lu commented 2 years ago

I also can't find the models in the website given in readme or in caffe model zoom, I download from openmmlab: https://download.openmmlab.com/pretrain/third_party/resnet50_caffe-788b5fa3.pth But I got the error about: Missing key(s) in state_dict: "conv1.weight", "bn1.weight" ...... then I modified the line 341 in dana.py from resnet.load_state_dict({k:v for k,v in state_dict.items() if k in resnet.state_dict()}) to resnet.load_state_dict(state_dict['state_dict'], False) Then I can run it, but it may not be the right way. Looking forward to author's reply

alphacyp commented 2 years ago

I also can't find the models in the website given in readme or in caffe model zoom, I download from openmmlab: https://download.openmmlab.com/pretrain/third_party/resnet50_caffe-788b5fa3.pth But I got the error about: Missing key(s) in state_dict: "conv1.weight", "bn1.weight" ...... then I modified the line 341 in dana.py from resnet.load_state_dict({k:v for k,v in state_dict.items() if k in resnet.state_dict()}) to resnet.load_state_dict(state_dict['state_dict'], False) Then I can run it, but it may not be the right way. Looking forward to author's reply

Thank you very much for your help. According to your tips, I have run successfully, but I only have two GPUs and run very slowly.

IceHowe commented 2 years ago

Can you tell me what GPU you have? And how much time it takes to train this model one epoch?

alphacyp commented 2 years ago

Can you tell me what GPU you have? And how much time it takes to train this model one epoch?

2 Nvidia GeForce GTX 2080Ti.Not so long