how to train keypoints model

menyifang / DCT-Net

Official implementation of "DCT-Net: Domain-Calibrated Translation for Portrait Stylization", SIGGRAPH 2022 (TOG); Multi-style cartoonization

Apache License 2.0

734 stars 74 forks source link

Closed kitterive closed 1 year ago

kitterive commented 1 year ago

I convert keypoints from tf pb model to onnx, and found the result is not same pb model result.

kitterive commented 1 year ago

可能是由于keypoints和 detect 两个模型都有traning flag的原因，在导出模型时，能去掉training flag这个输入吗，设为固定值False, 这样就一个输入，比较适合推理

kitterive commented 1 year ago

solved

cv-nlp commented 1 year ago

请问下您是怎么解决的呢？tf2onnx的时候能设置固定输入吗？有没有转换脚本呢