Export Frozen.pb with TF compatible nms_rotate
Zappytoes opened this issue · 0 comments
Zappytoes commented
I've exported a model from the FPN_Tensorflow_Rotation implementation, however, when I attempt run inference with tf.Session, I received the error:
ValueError: callback pyfunc_18 is not found
[[{node postprocess_fastrnn/Pyfunc_2}]]
[[{node postprocess_fastrnn/Pyfunc}]]
It looks like the nms for the rotated boxes was not written in TF and is not saveable to the graph, so was ignored.
Is there a compatible TF version of nms_rotate so the Frozen.pb file can be used?