Model size - I trained a model by the repo's code and recieved model with bigger size than the release model

edvardHua / PoseEstimationForMobile

:dancer: Real-time single person pose estimation for Android and iOS.

Apache License 2.0

1k stars 268 forks source link

Model size - I trained a model by the repo's code and recieved model with bigger size than the release model #145

Open yonatanbitton opened 4 years ago

yonatanbitton commented 4 years ago

Hello I take your original github implementation, with the mv2_cpm.cfg config file and the network_mv2_cpm.py. I trained a model for 24K steps and produced .pb model file.

Your .pb file here https://github.com/edvardHua/PoseEstimationForMobile/blob/master/release/cpm_model/model.pb is 2.6MB. My .pb is 5.4MB.

Why do we receive models with different size? How did you reach your 2.6MB?

Thanks

lzcchl commented 3 years ago

hello, did you solve this problem? I have same question? my tensorflow verison is 1.15