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

Convert ckpt to pb Success, Inference Fails #245

Open
github2016-yuan opened this issue Jun 17, 2020 · 2 comments
Open

Convert ckpt to pb Success, Inference Fails #245

github2016-yuan opened this issue Jun 17, 2020 · 2 comments

Comments

@github2016-yuan
Copy link

I have already converted ckpt to pb file. The size of pb file is 248.2MB. Then I want to use the pb file for inference. I find this sample code
https://github.com/github2016-yuan/tensorflow-yolov3/blob/master/image_demo.py
But it is different. I try but just get nothing. I know the reason but can not solve it.I think
https://github.com/wizyoung/YOLOv3_TensorFlow/blob/master/test_single_image.py
is also helpful when writting code of inference with pb file.

I realize some guys do not know what is ouput_node_names, so here are what I use.
output_node_names = ["yolov3/yolov3_head/feature_map_1", "yolov3/yolov3_head/feature_map_2", 'yolov3/yolov3_head/feature_map_3']

when inference with pb, the input_node_name is Placeholder and
return_elements = ["Placeholder:0", "yolov3/yolov3_head/feature_map_1:0", "yolov3/yolov3_head/feature_map_2:0", 'yolov3/yolov3_head/feature_map_3:0']

I check it and help it is useful for anyone who need it.
This repo is lack of pb file and inference with pb, I need a little help to finish it.

@giorgiomondauto
Copy link

Did you sort this issue out? Have you generated the .pbtxt file as well?

@qawsedrg
Copy link

generation of and inference with pb file is supported here, https://github.com/qawsedrg/yolov3_yolov4_tensorflow, as well as yolo tiny support and yolov4 support and other amazing features and is being maintained.

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

3 participants