Skip to content

Latest commit

 

History

History
26 lines (19 loc) · 1022 Bytes

README_CN.md

File metadata and controls

26 lines (19 loc) · 1022 Bytes

ONNX

deoldify-onnx-web

老照片上色功能,纯web端实现 英文文档中文文档

Latency Comparison

Latency Comparison

安装方式

你可以到发布页面,下载 deoldify.quant.onnx 文件,之后把它放到models目录即可

onnx 量化导出

我将原有的deoldify模型下载下来并进行了量化导出

import onnx
from onnxruntime.quantization import quantize_dynamic, QuantType

model_fp32 = './deoldify.onnx'
model_quant = './deoldify.quant.onnx'
quantized_model = quantize_dynamic(model_fp32, model_quant, weight_type=QuantType.QUInt8)

Credits

If you use any ideas from the papers or code in this repo, please consider citing the authors of deoldify. Lastly, if the ONNX versions helped you in any way, please also consider starring this repository.