Sep, 2022

使用知识蒸馏的对抗生成超分辨率在边缘设备上的应用

TL;DR本文提出了一个高效的生成对抗网络模型用于在需要可靠的视觉流来监视任务、处理远程操作或研究相关视觉细节的环境中支持机器人任务的单图像超分辨率,并通过定制的网络架构和模型量化使其在 CPU 和 Edge TPU 设备上达到高达 200 fps 的快速推断,同时进一步通过将其知识压缩至网络的较小版本来优化我们的模型,并与标准训练方法进行比较,得出显著的性能改善结果。我们的实验结果表明,相比于更重型的现有模型,我们的快速轻量模型在保持极高图像质量的同时,具有显著的优势,在带宽降级的图像传输方面的应用潜力也值得进一步挖掘。