see-- / keras-centernet

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

About convert pytorch model to keras or tensorflow #11

Open WangyiNTU opened 5 years ago

WangyiNTU commented 5 years ago

Thanks for your work. I am trying to train an hourglass architecture in official Pytorch implementation on different datasets. I plan to convert my Pytorch model to Keras and run it using your code. How did you convert the official hourglass model to .hdf5 files in your repository and could you share this code? Thanks! @see--