search

VITA-Group / EnlightenGAN

[IEEE TIP] "EnlightenGAN: Deep Light Enhancement without Paired Supervision" by Yifan Jiang, Xinyu Gong, Ding Liu, Yu Cheng, Chen Fang, Xiaohui Shen, Jianchao Yang, Pan Zhou, Zhangyang Wang
Other
890 stars 198 forks source link

Convert to ONNX #79

Closed make-j64 closed 3 years ago

make-j64 commented 3 years ago

Thanks for great work. I want to ask how to convert pretrained model to onnx format?

Thanks.

yifanjiang19 commented 3 years ago

I'm not very familiar with ONNX and feel free to pull request if you implement it.

yifanjiang19 commented 3 years ago

check here https://github.com/arsenyinfo/EnlightenGAN-inference