search

midasklr / RefineNet_TensorRT

TensorRT for RefineNet Segmentation
11 stars 6 forks source link

output layer #1

Open guods opened 4 years ago

guods commented 4 years ago

would you like to tell me about the name of output layer?

midasklr commented 4 years ago

I recommend u convert to onnx with this scrip:https://github.com/midasklr/RefineNet/blob/master/src/onnx_export.py I trained with pytorch 1.3.1 but export onnx model with Pytorch 0.41

guods commented 4 years ago

@midasklr I have a question, usually converted to onnx files requires given input and output, Why is the output not given in "onnx_export.py"?