search

see-- / keras-centernet

A Keras implementation of CenterNet with pre-trained model (unofficial)
MIT License
337 stars 84 forks source link

Pytorch to keras weight mapping #18

Closed PushparajaMurugan closed 4 years ago

PushparajaMurugan commented 5 years ago

Hi, YOu have done a grt job. Could you pls share the scripts for mapping pytoch weights to keras framework. Pytorch2keras is not working propoerly for the arch.

Thanks

see-- commented 5 years ago

Thanks for reporting that pytorch2keras does not work here. Could you check if #16 solves your issue? Let me know if it works, so that I can merge or recheck.

Edit: Do you mean the pull-request or the pytorch2keras repository? https://github.com/nerox8664/pytorch2keras

PushparajaMurugan commented 5 years ago

Hi Its the" pytorch2keras repository".

I have tested the other models using that repo and was unable to run this repo on your arch. The problem that I have faced is, it always gives the same probability scores and class number. I will check #16 and let you know the outcomes. Thanks,