Hi everyone. I need to deploy ssd object detection model on android, but since torch ops does not available on android. I need to trace decode function (https://github.com/amdegroot/ssd.pytorch/blob/master/layers/functions/detection.py) to decode detection result. but I do not manage to trace it. Does any one succeed to trace the decode part ?
I tried to trace the function but I got the following error:
Compiled from code export_decode_model.py(121): decode
export_decode_model.py(129): forward
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py(525): _slow_forward
/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py(539): call
/usr/local/lib/python3.6/dist-packages/torch/jit/init.py(997): trace_module
/usr/local/lib/python3.6/dist-packages/torch/jit/init.py(858): trace
export_decode_model.py(142):
The above operation failed in interpreter, with the following stack trace: