search

roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 567 forks source link

What is "detectron_weight_mapping" for? #154

Closed bowenc0221 closed 5 years ago

bowenc0221 commented 5 years ago

Could anyone explain what "detectron_weight_mapping" for?
It seems it is only used to "filter" some weights when loading a model in here.

Thanks.

achalddave commented 5 years ago

I think it's primarily to map the caffe2 detectron model keys to the pytorch detectron model parameter names, e.g. here and throughout here