/ONNX_GRAY2RGB

pytorch model -> ONNX model -> TNN model

Primary LanguagePython

灰度图上色模型

本仓库使用 pytorch 构建GAN模型实现灰度图上色的功能,开源了ONNX模型

执行方式:python test_onnx.py

利用移动端深度学习推理框架 TNN 执行iphone端上的前向推理,效果图如下

图片名称

图片名称