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

How can I read input and output node from frozen inference model (.pb)? #25

Open
K-M-Ibrahim-Khalilullah opened this issue Nov 22, 2019 · 2 comments

Comments

@K-M-Ibrahim-Khalilullah

@bendangnuksung
Thanks for your great repository. it helps me lot.

Would you suggest me please How can I read input and output node/tensor from frozen inference graph (.pb)?
Thanks again

@bendangnuksung
Copy link
Owner

Hi @IbrahimLearning
How is it different from Issues #9? Judging from your comment on #9 you seem to have successfully able to use the frozen model.

@K-M-Ibrahim-Khalilullah
Copy link
Author

K-M-Ibrahim-Khalilullah commented Nov 22, 2019

Hi @IbrahimLearning
How is it different from Issues #9? Judging from your comment on #9 you seem to have successfully able to use the frozen model.

@bendangnuksung
Thanks for your reply
Yes I used the frozen model using forward.py, what you provided. It was used for detection. In the internal code, the following segment mentions output nodes:
self.outputs = { 'detection': 'mrcnn_detection_4/Reshape_1', 'class': 'mrcnn_class_4/Reshape_1', 'box': 'mrcnn_bbox_4/Reshape', 'mask': 'mrcnn_mask_4/Reshape_1'}

But my problem is to display or read the node input and output from .pb file, where I don't know before the name of the model.
Is there any clue please?

Thanks

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

2 participants