聲明：本博文做了該代碼的測(cè)試分享蒲牧，敬請(qǐng)查閱；
墨理學(xué)AI.png
本文巡莹、節(jié)選自 圖像修復(fù)專欄
圖像修復(fù)新的創(chuàng)作思路：CVPR 2021司志、代碼測(cè)評(píng)
?? 基本信息
- Restoring Extremely Dark Images in Real Time
- 實(shí)時(shí)恢復(fù)極暗的圖像
- https://github.com/MohitLamba94/Restoring-Extremely-Dark-Images-In-Real-Time
- Lamba_Restoring_Extremely_Dark_Images_in_Real_Time_CVPR_2021_paper.pdf
image.png
該論文所致力于解決的問(wèn)題
- 極暗圖像變?yōu)?亮的圖像（圖像修復(fù)）
- 從而提升（解決）極暗圖像的目標(biāo)檢測(cè)問(wèn)題
image.png
?? 下載代碼
方式一（網(wǎng)絡(luò)受限、可能下載失斀嫡）
git clone https://github.com/MohitLamba94/Restoring-Extremely-Dark-Images-In-Real-Time.git
方式二（手動(dòng)下載骂远、copy 到服務(wù)器、解壓即可）
- 解壓命令钉鸯，例如
unzip Restoring-Extremely-Dark-Images-In-Real-Time-main.zip
image.png
?? 環(huán)境搭建
image.png
激活一個(gè) PyTorch 1.4 的已有環(huán)境（我的博文已經(jīng)安裝過(guò)很多個(gè)版本吧史、此處不再重復(fù)贅述）
- 不熟悉的可以去看我的這個(gè)博文
- Linux下cuda10.0安裝Pytorch和Torchvision——啥版本都能裝
conda activate torch14
# 安裝一些我的環(huán)境運(yùn)行該代碼、缺少的庫(kù)
pip install rawpy
pip install ptflops
?? Demo 測(cè)試運(yùn)行
cd Restoring-Extremely-Dark-Images-In-Real-Time
python demo.py
- 運(yùn)行輸出如下
python demo.py
# GPU 占用 不會(huì)很高
...... Loading all files to CPU RAM
Image No.: 1, Amplification_m=1: 53.080570220947266
Image No.: 2, Amplification_m=1: 22.907602310180664
Image No.: 3, Amplification_m=1: 45.878238677978516
Files loaded to CPU RAM......
Network parameters : 784768
Device on GPU: True
Restored images saved in DEMO_RESTORED_IMAGES directory
image.png
?? 時(shí)間內(nèi)存復(fù)雜度測(cè)評(píng)
Measure Time-Memory Complexity
python time_complexity.py- 運(yùn)行效果如下
python time_complexity.py
---Our Model parameters : 784768
---SID model parameters : 7760748
Computational complexity of Our model for a 8MP image: 41.38 GMac
Computational complexity of SID model for a 8MP image: 440.46 GMac
Beginning Warmup...
Time taken by our model on CPU for 8MP image : 1.0671975135803222 seconds
Time taken by SID model on CPU for 8MP image : 8.417949628829955 seconds
?? 訓(xùn)練
訓(xùn)練部分唠雕、參考
train_test_ours/train.py即可
暫不展開贸营、以后如果項(xiàng)目中用到、有需要再補(bǔ)充
image.png
?? 附源碼+論文
這些其實(shí)都很好下載岩睁、代碼此次也沒(méi)有改動(dòng)钞脂、即可順利運(yùn)行
鏈接：https://pan.baidu.com/s/129MAPqMJtNp1v57gHZDMCA
提取碼：moli
?? 這篇文章可以帶給我們的思考
翻譯部分、參考鏈接
image.png
特色
- 輕量化
- 圖像修復(fù)+目標(biāo)檢測(cè) 結(jié)合
- 突出解決實(shí)際模型部署落地中的困難點(diǎn)：
- 修復(fù)網(wǎng)絡(luò)捕儒、單張圖像推理速度慢
- 暗圖像目標(biāo)檢測(cè)存在困難
這是一個(gè) 圖像修復(fù)+目標(biāo)檢測(cè) 結(jié)合 針對(duì) 實(shí)際落地存在的現(xiàn)實(shí)困難 提出的解決方案冰啃，也許可以成為我們小伙伴，創(chuàng)作一篇文章（Paper）的靈感基石
image.png
???? 墨理學(xué)AI
博主簡(jiǎn)介：軟件工程專業(yè)刘莹、2020年碩士畢業(yè)阎毅、總計(jì) 10 w 讀者 粉絲
- ?? 計(jì)算機(jī)視覺(jué)：超分重建、圖像修復(fù)点弯、目標(biāo)檢測(cè)扇调、風(fēng)格遷移 等領(lǐng)域 稍有所學(xué)
- ?? AI 工程化：Ncnn、MNN抢肛、TensorRT 正在 學(xué)習(xí)
- ?? C++狼钮、Python碳柱、Java 略懂一二
喜歡請(qǐng)關(guān)注 墨理學(xué)AI
取經(jīng)路上，讓墨理學(xué)AI 陪你暢享更多有趣AI
