hi, i'm trying to convert SSD pytorch model to DLC using SNPE version 1.54.2.
my SSD model has two output node, "concatenation_0_0" for bbox and "concatenation_1_0" for class score
I can convert model with one of two output node, but I can't convert with both of them.
for example,
snpe-pytorch-to-dlc --input_network ./ssd_300_traced.jit.pt -d x 1,3,300,300 --out_node cancatenation_0_0
produces cancatenation_0_0.raw and
snpe-pytorch-to-dlc --input_network ./ssd_300_traced.jit.pt -d x 1,3,300,300 --out_node concatenation_1_0
produces concatenation_1_0 .raw
but,
snpe-pytorch-to-dlc --input_network ./ssd_300_traced.jit.pt -d x 1,3,300,300 --out_node concatenation_0_0 --out_node cancatenation_1_0
produces only one output node, node cancatenation_1_0.raw
is there any solution for this issue?
Dear customer,
Thanks for your efforts in our products.
Please just remove the --out_node inside SNPE ONNX converter. And then write the out_node with #output to your input list file while you run on final devices. Like as below.
#output1 output2
input:=1.raw
That may resolve your questions.
BR.
Wei