staillyd / cosine_metric_learning Goto Github PK

hi, i have converted the ckpt files to pb(frozen tensorflow model) with freeze_graph.py(tensorflow tools), then i have a test, thers is one error:
Traceback (most recent call last):
File "../work/StereoCamera/pb_test/tf_inference.py", line 53, in
OUTS = sess.run([tensor_output], feed_dict={tensor_input: image_np_expanded})
File "../.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 950, in run
run_metadata_ptr)
File "../.local/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1126, in _run
subfeed_dtype = subfeed_t.dtype.as_numpy_dtype
File "../.local/lib/python3.6/site-packages/tensorflow/python/framework/dtypes.py", line 126, in as_numpy_dtype
return _TF_TO_NP[self._type_enum]
KeyError: 20
have you met the same error, how can i solved it? Regards!