microsoft / Bringing-Old-Photos-Back-to-Life

Bringing Old Photo Back to Life (CVPR 2020 oral)
https://arxiv.org/abs/2004.09484
MIT License
15.15k stars 2k forks source link

Inference on CPP code #229

Closed FrancescaCi closed 2 years ago

FrancescaCi commented 2 years ago

Hi, I am very interesting on your project, I would like to know if it is possible to inference the pretrained model directly on a test set of images but using CPP implementation. If yes, can I ask some help for some scratch about it? Thank you for your feedback. Best

tongchangD commented 2 years ago

I was building this and encountered the following problem when PTH was transferred to ONNX.

in Face_Enhancement.

RuntimeError: Unsupported: ONNX export of batch_norm for unknown channel size.
FrancescaCi commented 2 years ago

Thank you so much for your suggestion! A question, if you try the model for global-restoration only to apply directly on images, is the result the same?

tongchangD commented 2 years ago

haven't tried