Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Model freezing #28

Closed
Abdulazizbek opened this issue Sep 5, 2019 · 9 comments
Closed

Model freezing #28

Abdulazizbek opened this issue Sep 5, 2019 · 9 comments

Comments

@Abdulazizbek
Copy link

Have you tried to freeze trained model from checkpoints to .pb file ( converting .ckpt, .meta, .index inside checkpoints folder to frozen model .pb) in a end-to-end way?
Thank you beforehand

@Pay20Y
Copy link
Owner

Pay20Y commented Sep 5, 2019

Sorry I have not tried it. Since I'm still doing experiments to reimplement the effect in paper. When I make it I'll release a frozen model.

@bharatsubedi
Copy link

I tried to freeze model but I faced one issue for model freezing. For detection session output of detection is f_score and f_geometry is in a graph so it will easily convert but in cases of recognition session dense_decode which is the output node of recognition is not inside the graph. do you know how to solve and freeze model? could you provide some hints on how can I solve this issue?

@Pay20Y
Copy link
Owner

Pay20Y commented Sep 6, 2019

Maybe reimplement the NMS and the postprocess of detection(generating box from f_geometry) with tensorflow api.

@bharatsubedi
Copy link

could you explain more about model freezing? I tried different way but fails to get dense_decode output inside recog_decode session.

@Pay20Y
Copy link
Owner

Pay20Y commented Sep 12, 2019

Sorry, I have never tried to freeze the model.

@lerndeep
Copy link

hello pay20Y,

I am also doing model freezing. I got output dense_decode but when i tried to run a session for recognition feeding input value for input_box_mask I got the issue.
image
Due to shape() issue I could not feed value into the session. could you provide some information?

@Pay20Y
Copy link
Owner

Pay20Y commented Sep 19, 2019

@lerndeep input_box_mask is a list of 1D tensor. It is used to indicate which feature map that rois belonging to. So if you run with batch size of 1 the feeding input value is [0] * num_of_rois

@lerndeep
Copy link

Thank you for your reply. Now I solved the tensor size issue but I got another Issue "input_box_masks_.
image
This is my detail implementation code.

@Alek-dr
Copy link

Alek-dr commented Jul 22, 2020

Have somebody success in freezing?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

5 participants