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

Loading detectron weights Error #170

Open B2Gdevs opened 5 years ago

B2Gdevs commented 5 years ago

Error : " if 'blobs' in src_blobs: TypeError: argument of type 'int' is not iterable"

I have been using detectron, but I have been using the load_ckpt method. I have been trying to do the methods proposed in other issues to modify the output layer, however this uses the load_detectron_weights method, which gives me the error above. I am needing to know how to solve this.