onnx input name and output name as 'input' and 'output'

Author: ersheng-ai

demo_pytorch2onnx.py

@@ -31,7 +31,7 @@ def transform_to_onnx(weight_file, batch_size, n_classes, IN_IMAGE_H, IN_IMAGE_W
                       export_params=True,
                       opset_version=11,
                       do_constant_folding=True,
-                      # input_names=['input'], output_names=['output_1', 'output_2', 'output_3'],
+                      input_names=['input'], output_names=['output'],
                       dynamic_axes=None)
 
     print('Onnx model exporting done')

tool/darknet2onnx.py

@@ -24,7 +24,7 @@ def transform_to_onnx(cfgfile, weightfile, batch_size=1):
                       export_params=True,
                       opset_version=11,
                       do_constant_folding=True,
-                      # input_names=['input'], output_names=['output_1', 'output_2', 'output_3'],
+                      input_names=['input'], output_names=['output'],
                       dynamic_axes=None)
 
     print('Onnx model exporting done')