search

zengarden / light_head_rcnn

Light-Head R-CNN
834 stars 222 forks source link

No op named PSAlignPool in defined operations #74

Open YoungDav opened 5 years ago

YoungDav commented 5 years ago

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 commented 5 years ago

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 commented 5 years ago

@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。