search

PINTO0309 / PINTO_model_zoo

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.
https://qiita.com/PINTO
MIT License
3.49k stars 566 forks source link

RoI Tanh-polar Transformer Network for Face Parsing ONNX conversion #353

Closed harisreedhar closed 11 months ago

harisreedhar commented 1 year ago

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

PINTO0309 commented 1 year ago

WIP

https://github.com/PINTO0309/face_parsing image

harisreedhar commented 1 year ago

WIP

https://github.com/PINTO0309/face_parsing image

Any update?

PINTO0309 commented 11 months ago

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.

harisreedhar commented 11 months ago

Ok. Thanks for trying

ykk648 commented 2 months ago

Hi, are there any new possibilities to transform the tanh-polar op after one year?