Path: blob/master/site/zh-cn/lite/models/super_resolution/overview.ipynb
25118 views
Kernel: Python 3
Copyright 2020 The TensorFlow Authors.
In [ ]:
使用 TensorFlow Lite 实现超分辨率
文本特征向量
从低分辨率图像恢复高分辨率图像的任务通常被称为单图像超分辨率 (SISR)。
这里使用的模型是 ESRGAN(ESRGAN:增强超分辨率生成对抗网络)。我们将使用 TensorFlow Lite 对预训练的模型进行推断。
该 TFLite 模型从托管在 TF Hub 上的此实现转换而来。请注意,我们转换的模型会将 50x50 低分辨率图像上采样为 200x200 高分辨率图像(比例因子 = 4)。如果需要不同的输入大小或比例因子,则需要重新转换或重新训练原始模型。
安装
我们先安装所需的库。
In [ ]:
导入依赖项。
In [ ]:
下载并转换 ESRGAN 模型
In [ ]:
下载测试图像(昆虫头部)。
In [ ]:
使用 TensorFlow Lite 生成超分辨率图像
In [ ]:
呈现结果
In [ ]:
性能基准
性能基准数值使用此处所述工具生成。
模型名称 | 模型大小 | 设备 | CPU | GPU |
---|---|---|---|---|
super resolution (ESRGAN) | 4.8 Mb | Pixel 3 | 586.8ms* | 128.6ms |
Pixel 4 | 385.1ms* | 130.3ms |
*使用 4 个线程。