RoI Tanh-polar Transformer Network for Face Parsing ONNX conversion
harisreedhar opened this issue · comments
Issue Type
Feature Request
OS
Ubuntu
OS architecture
x86_64
Programming Language
Python
Framework
ONNX
Model name and Weights/Checkpoints URL
rtnet50-fcn-11.torch
rtnet50-fcn-14.torch
rtnet101-fcn-14.torch
resnet50-fcn-14.torch
deeplabv3plus-14.torch
Description
Would be nice to port this pytorch Face-Parsing Model to modelzoo as onnx.
Relevant Log Output
No response
URL or source code for simple inference testing code
https://github.com/hhj1897/face_parsing/blob/master/ibug/face_parsing/parser.py
WIP
Any update?
Tanh-polar is quite difficult to convert to ONNX; many operations cannot be exported to ONNX. It is very easy to convert the part except for tanh-polar.
Ok. Thanks for trying