论文标题
MicroISP:在移动设备上处理32MP的照片,并深入学习
MicroISP: Processing 32MP Photos on Mobile Devices with Deep Learning
论文作者
论文摘要
尽管与传统的ISP系统相比,基于神经网络的照片处理解决方案可以提供更好的图像质量,但由于它们非常高的计算复杂性,它们在移动设备上的应用仍然非常有限。在本文中,我们提出了一个新颖的MicroISP模型,考虑到其计算和内存限制,专门为边缘设备设计。所提出的解决方案能够使用标准移动ML库在最近的智能手机上处理多达32MP的照片,并且需要少于1秒钟的时间才能执行推理,而对于FullHD图像,它可以实现实时性能。该模型的体系结构是灵活的,可以将其复杂性调整为不同计算功率的设备。为了评估模型的性能,我们收集了一种新颖的Fujifilm Ultraisp数据集,该数据集由数千张配对照片组成,这些照片用普通的移动摄像头传感器和专业的102MP中型富士Fujifilm GFX100相机捕获。该实验表明,尽管具有紧凑的尺寸,MicroISP模型还是能够提供可比或更好的视觉结果,而传统的移动ISP系统则表现出色,同时胜过先前提出的有效的基于深度学习的解决方案。最后,该模型还与最新的移动AI加速器兼容,可以在智能手机NPU和APU上实现良好的运行时和低功耗。代码,数据集和预培训模型可在项目网站上找到:https://people.ee.eethz.ch/~ihnatova/microisp.html
While neural networks-based photo processing solutions can provide a better image quality compared to the traditional ISP systems, their application to mobile devices is still very limited due to their very high computational complexity. In this paper, we present a novel MicroISP model designed specifically for edge devices, taking into account their computational and memory limitations. The proposed solution is capable of processing up to 32MP photos on recent smartphones using the standard mobile ML libraries and requiring less than 1 second to perform the inference, while for FullHD images it achieves real-time performance. The architecture of the model is flexible, allowing to adjust its complexity to devices of different computational power. To evaluate the performance of the model, we collected a novel Fujifilm UltraISP dataset consisting of thousands of paired photos captured with a normal mobile camera sensor and a professional 102MP medium-format FujiFilm GFX100 camera. The experiments demonstrated that, despite its compact size, the MicroISP model is able to provide comparable or better visual results than the traditional mobile ISP systems, while outperforming the previously proposed efficient deep learning based solutions. Finally, this model is also compatible with the latest mobile AI accelerators, achieving good runtime and low power consumption on smartphone NPUs and APUs. The code, dataset and pre-trained models are available on the project website: https://people.ee.ethz.ch/~ihnatova/microisp.html