ICPR 2022 | PyNet-V2 Mobile:分組殘差+通道/空間雙注意力,手機端12MP RAW照片1.5秒直出!
這篇論文最有意思的地方,不是把注意力機制簡單地塞進網絡,而是在移動端 AI 加速器只支持 101 種算子、RAM 極其有限的苛刻約束下,用"分組殘差 + 通道/空間雙注意力"把整個 RAW 到 RGB 的 ISP 流程壓進 3.6MB 的模型里——12MP 照片端到端直出,畫質卻逼近中畫幅專業(yè)相機。論文: PyNet-V2 Mobile: Efficient On-Device Photo Processing With Neural Networks作者: Andrey Ignatov, Grigory Malivenko, Radu Timofte, Yu Tseng, Yu-Syuan Xu, Po-Hsiang Yu, Cheng-Ming Chiang, Hsien-Kai Kuo, Min-Hung Chen, Chia-Ming Cheng, Luc Van Gool發(fā)表單位: ETH Zurich / AI Witchlabs / University of Würzburg / MediaTek Inc.發(fā)表: ICPR 2022論文鏈接: https://arxiv.org/abs/2211.06263代碼鏈接: https://github.com/gmalivenko/PyNET-v2一、引言手機已成為普通人拍照的主要設備,但小尺寸傳感器在動態(tài)范圍、噪點、色彩上都遠遜于專業(yè)相機。傳統(tǒng) ISP(圖像信號處理)流水線依賴手工設計的去馬賽克、降噪、銳化等模塊,已很難再榨出多少畫質提升。于是"用深度學習直接替代整個 ISP"(learned ISP)成為熱點方向:輸入 RAW 數(shù)據,端到端輸出高質量 RGB 照片。此前的代表性工作 PyNET [32] 已證明這條路可行,但它的模型高達數(shù)百 MB、計算量巨大,一張 12MP 照片在手機 GPU 上推理直接 OOM(內存溢出),根本無法落地。而另一類為速度設計的輕量模型(FSRCNN、SmallNet 等)雖然跑得快,畫質卻明顯拉胯,噪聲抑制和色彩還原基本不可用。本文提出的PyNet-V2 Mobile正是要填上這個鴻溝:在移動 AI 加速器的算子約束、內存約束、體積約束下,實現(xiàn)高畫質 + 低延遲的端到端 RAW 照片處理。二、核心動機移動端端到端 ISP 的三重約束任務復雜度:模型既要完成全局調整(亮度、白平衡、色彩還原),又要完成局部處理(紋理增強、降噪、2 倍超分)。ResNet/U-Net 等單一結構只能顧一頭。硬件算子限制:Android NNAPI 1.2 只支持有限算子(最新系統(tǒng)最多 101 種,老系統(tǒng)甚至不足 28 種),并且 RAM 極小——FullHD 推理可用內存往往只有幾百 MB。體積限制:模型需要隨相機應用一起打包,數(shù)百 MB 的模型不可接受。關鍵觀察原版 PyNET 之所以"重",主要因為:每個尺度通道數(shù)逐級翻倍(高達 512 通道)、大量大卷積核與轉置卷積、以及缺少輕量化的全局建模手段。作者觀察到:移動端模型應當在尺度變大時"越用越省"——高層通道數(shù)減半,并僅用 3×3 卷積 + 分組卷積 + 輕量注意力,就能在算子允許的范圍內同時獲得全局調整與局部重建能力。三、方法3.1 模塊整體設計PyNet-V2 Mobile 整體架構采用倒金字塔(inverted pyramid)三級尺度結構,數(shù)據流為:輸入 RAW RGBG Bayer 數(shù)據,經space-to-depth拆分為 4 個通道(對應 B/Gb/R/Gr);依次經過 3 個尺度(Level 3 → Level 2 → Level 1)處理,每個尺度完成一次"下采樣、殘差處理、輸出預測";低尺度特征經雙線性上采樣后與高尺度特征拼接,實現(xiàn)逐級細化;每個尺度末端輸出tanh\text{tanh}tanh激活的 RGB 結果,最終取最高分辨率輸出。通道配置為 32 → 64 → 128(每升高一個尺度翻倍),但相比原版 PyNET 每尺度通道數(shù)整體減半,3×3 卷積 + PReLU 為基本操作,最后輸出層用 Tanh 將結果映射到(?1,1)(-1, 1)(?1,1):out=0.58?tanh?(z)+0.5\text{out} = 0.58 \cdot \tanh(z) + 0.5out=0.58?tanh(z)+0.53.2 關鍵操作:Space-to-depth 輸入輸入為原始 Bayer 數(shù)據I∈RH×WI \in \mathbb{R}^{H \times W}I∈RH×W,通過 space-to-depth 操作拆成 4 通道:IBayer→space-to-depthI^∈RH2×W2×4I_{\text{Bayer}} \xrightarrow{\text{space-to-depth}} \hat{I} \in \mathbb{R}^{\frac{H}{2} \times \frac{W}{2} \times 4}IBayer?space-to-depth?I^∈R2H?×2W?×4chB=I[1::2,1::2],chGb=I[0::2,1::2],chR=I[0::2,0::2],chGr=I[1::2,0::2]\text{ch}_B = I[1::2, 1::2], \quad \text{ch}_{Gb} = I[0::2, 1::2], \quad \text{ch}_R = I[0::2, 0::2], \quad \text{ch}_{Gr} = I[1::2, 0::2]chB?=I[1::2,1::2],chGb?=I[0::2,1::2],chR?=I[0::2,0::2],chGr?=I[1::2,0::2]這樣既保留了 Bayer 的顏色相位信息,又天然完成了 2 倍下采樣,與末端 depth-to-space 上采樣形成對稱結構。3.3 核心子模塊(1) 分組殘差塊(Grouped Residual Block)將輸入按通道拆成GGG組(2~4 組),各組并行執(zhí)行nnn次 3×3 卷積(偶數(shù)分組額外施加實例歸一化 InstanceNorm),再拼接、經 1×1 卷積恢復通道數(shù)后與輸入相加:xout=x+f1×1([ gi+f3×3(gi) ]i=1G),gi=x[:,:,:,iCG:(i+1)CG]x_{\text{out}} = x + f_{1\times1}\Big(\Big[\, g_i + f_{3\times3}(g_i) \,\Big]_{i=1}^{G}\Big), \quad g_i = x\Big[:, :, :, \tfrac{iC}{G}:\tfrac{(i+1)C}{G}\Big]x

相關新聞

DFRC系統(tǒng)波束成形設計與Matlab仿真實踐

DFRC系統(tǒng)波束成形設計與Matlab仿真實踐

1. 項目背景與核心價值 雙功能雷達通信系統(tǒng)(Dual-Function Radar-Communication, DFRC)是當前無線通信與雷達探測融合的前沿研究方向。我在實際工程中發(fā)現(xiàn),傳統(tǒng)系統(tǒng)往往需要獨立部署雷達和通信設備,導致頻譜資源緊張、硬件成本高昂…

2026/8/3 8:48:39 閱讀更多
OpenGL著色器類封裝:從原理到實踐,提升圖形編程效率

OpenGL著色器類封裝:從原理到實踐,提升圖形編程效率

1. 項目概述:為什么我們需要封裝著色器類? 如果你剛開始接觸C和OpenGL,大概率是從畫一個三角形開始的。跟著教程,你會寫一大堆代碼:創(chuàng)建窗口、初始化GLAD、定義頂點數(shù)據、編譯頂點和片段著色器、鏈接成著色器程序&…

2026/8/3 8:48:39 閱讀更多
GitHub私有倉庫SSH訪問配置全流程指南

GitHub私有倉庫SSH訪問配置全流程指南

1. GitHub 私有倉庫SSH訪問配置全流程指南作為開發(fā)者日常工作的剛需,SSH密鑰訪問GitHub私有倉庫的配置看似簡單,實際暗藏不少平臺差異性和配置細節(jié)。我在為團隊制定標準化操作流程時,發(fā)現(xiàn)即便是經驗豐富的工程師,也常會在密鑰權限…

2026/8/3 9:48:41 閱讀更多
綜合能源系統(tǒng)智能調度:CSP、ORC與P2G的Matlab優(yōu)化實踐

綜合能源系統(tǒng)智能調度:CSP、ORC與P2G的Matlab優(yōu)化實踐

1. 項目概述:綜合能源系統(tǒng)的智能調度方案這個項目解決的是現(xiàn)代能源系統(tǒng)中一個關鍵痛點——如何高效協(xié)調多種異質能源的調度問題。我們構建了一個包含光熱電站(CSP)、有機朗肯循環(huán)(ORC)和電轉氣(P2G)技術的綜合能源系統(tǒng),通過Matlab實現(xiàn)優(yōu)化調度算法。這種…

2026/8/3 9:48:41 閱讀更多
OpenClaw升級安裝教程,TopClaw3分鐘自動更新滿血內核

OpenClaw升級安裝教程,TopClaw3分鐘自動更新滿血內核

升級OpenClaw前,先搞清楚這幾點用過OpenClaw的朋友應該都有這種感覺——它確實能干,但每次版本一更新,手動折騰依賴、配置環(huán)境、還要擔心兼容性問題,真的挺磨人。我之前也踩過不少坑,比如升級到一半報錯、舊配置失效、…

2026/8/3 9:48:41 閱讀更多
C# 周記 :從集合字典到 IO 多線程的踩坑實錄

C# 周記 :從集合字典到 IO 多線程的踩坑實錄

這一周的學習進度可以說是“起飛”兼“翻車”并存。從抽象類一路狂飆到的 IO 與文件操作,順帶還和多線程(Multithreading)剛了正面。代碼邏輯越來越復雜,遇到的 Bug 也越來越底層。今天總結一下這周的核心知識點,以及那…

2026/8/3 9:48:41 閱讀更多
3分鐘搞定!QQ空間歷史說說完整備份終極指南

3分鐘搞定!QQ空間歷史說說完整備份終極指南

3分鐘搞定!QQ空間歷史說說完整備份終極指南 【免費下載鏈接】GetQzonehistory 獲取QQ空間發(fā)布的歷史說說 項目地址: https://gitcode.com/GitHub_Trending/ge/GetQzonehistory 你是否曾想過,那些年發(fā)過的QQ空間說說,那些記錄青春的文字…

2026/8/2 0:04:01 閱讀更多
AMAT 0100-02186 I/O 分配 PCB

AMAT 0100-02186 I/O 分配 PCB

AMAT 0100-02186 I/O分配PCB板是應用材料(Applied Materials)公司生產的一款用于半導體設備的I/O信號分配電路板。該型號(0100-02186)的核心特點如下:專用于Endura等半導體工藝腔室。集成信號路由與分配功能。連接控制…

2026/8/2 2:51:21 閱讀更多
Nissei Corp FFMN-32L-10-T0 40AX 三相異步電動機

Nissei Corp FFMN-32L-10-T0 40AX 三相異步電動機

Nissei Corp FFMN-32L-10-T0 40AX 三相異步電動機是日本日清(Nissei)品牌的一款工業(yè)用三相異步電機,適用于自動化設備及通用機械驅動。該型號(FFMN-32L-10-T0 40AX)的核心特點如下:三相交流異步電動機。額定…

2026/8/2 2:52:49 閱讀更多