search

xingyizhou / CenterNet2

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

Can't load checkpoint from detectron2://ImageNetPretrained/MSRA/R-50.pkl #49

Open XYSC opened 3 years ago

XYSC commented 3 years ago

Hello, I am very interested in your work. I encountered a few problems while training my samples. Because my server can't access the external network, I encountered such a problem:

image

Then I found the corresponding R-50.pkl download link in “modelzoo.md” by searching. After downloading it, I replaced "WEIGHTS:“ in “Base-CenterNet2.yaml” with the local path, and then reported this error:

image

I don’t know how to modify the corresponding code... Should I use the local address of the .pkl to replace the online URL? Or how should I do? Looking forward to your reply, thank you very much

xingyizhou commented 3 years ago

Sorry for the delayed response. Yes, you can manually download the file from the URL and load it locally by modifying MODEL.WEIGHTS.