I have a tensorflow model with mutiple output, how to convert tensorflow model with mutiple output to UFF. How to write the OUTPUT_NAMES in uff_model = uff.from_tensorflow(frozen_graph, OUTPUT_NAMES) and how to write parser.register_output()?

Hi, For example, uff_model = uff.from_tensorflow(tf_model, ['out1','out2','out3']) ... parser.register_output("out1") parser.register_output("out2") parser.register_output("out3") ... Thanks.

how to convert tensorflow model with mutiple output to UFF

Autonomous Machines Jetson & Embedded Systems Jetson TX2

samjith888 August 27, 2020, 7:48am 13

Please look into this thread