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

No op named PSAlignPool in defined operations #74

Open
YoungDav opened this issue May 20, 2019 · 2 comments
Open

No op named PSAlignPool in defined operations #74

YoungDav opened this issue May 20, 2019 · 2 comments

Comments

@YoungDav
Copy link

hi:
i trans model.ckpt to .pb according to https://github.com/zengarden/light_head_rcnn/issues/66
but when i start inference function , it raise such error:

Traceback (most recent call last):
File "pb_inference.py", line 36, in
model_fn = DetectionModel()
File "pb_inference.py", line 16, in init
tf.import_graph_def(od_graph_def, name='')
File "/home/luban/miniconda3/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 316, in new_func
return func(*args, **kwargs)
File "/home/luban/miniconda3/lib/python3.6/site-packages/tensorflow/python/framework/importer.py", line 541, in import_graph_def
raise ValueError('No op named %s in defined operations.' % node.op)
ValueError: No op named PSAlignPool in defined operations.

@YoungDav
Copy link
Author

i think may solve it:
add this line before tf.import_graph_def(od_graph_def, name='')

PSAlignPool = tf.load_op_library('lib/lib_kernel/lib_psalign_pooling/psalign_pooling.so')

@Willis90
Copy link

Willis90 commented Jun 13, 2019

@YoungDav hi I want to freeze CKPT into pb,But I do not know what's name of output node?What's the output node name do you use ?I use res_v1_101 for my backbone net.Can you give some help?Thank you。

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