SJTU-Thinklab-Det/DOTA-DOAI

Export Frozen.pb with TF compatible nms_rotate

Zappytoes opened this issue · 0 comments

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?