DeepMosaics

這是一個通過深度學習自動的為圖片/視頻添加馬賽克,或消除馬賽克的項目.

它基于“語義分割”以及“圖像翻譯”.

更多例子

原始

自動打碼

自動去碼

馬賽克圖片

DeepCreamPy

ours

風格轉換

原始

梵高風格

轉化為冬天

一個有意思的嘗試:香蕉君♂貓

如何運行

可以通過我們預編譯好的二進制包或源代碼運行.

預編譯的程序包

對于Windows用戶,我們提供了包含GUI界面的免安裝軟件包.

可以通過下面兩種方式進行下載: [Google Drive] [百度云,提取碼1x0a]

注意事項:

程序的運行要求在64位Windows操作系統,我僅在Windows10運行過,其他版本暫未經過測試

請根據需求選擇合適的預訓練模型進行測試，不同的預期訓練模型具有不同的效果.[預訓練模型介紹]

運行時間取決于電腦性能,對于視頻文件,我們建議使用源碼并在GPU上運行.

如果輸出的視頻無法播放,這邊建議您嘗試potplayer.

相比于源碼,該版本的更新將會延后.

通過源代碼運行

前提要求

Linux, Mac OS, Windows

Python 3.6+

CPU or NVIDIA GPU + CUDA CuDNN

Python依賴項

代碼依賴于opencv-python以及 torchvision,可有通過pip install 進行安裝.

克隆源代碼

git clone https://github.com/HypoX64/DeepMosaics

cdDeepMosaics

下載預訓練模型

可以通過以下兩種方法下載預訓練模型,并將他們置于'./pretrained_models'文件夾中.

[Google Drive] [百度云,提取碼1x0a]

[預訓練模型介紹]

簡單的例子

為視頻添加馬賽克,例子中認為臉是需要打碼的區域 ,可以通過切換預訓練模型切換自動打碼區域(輸出結果將儲存到 './result')

python deepmosaic.py --media_path ./imgs/ruoruo.jpg --model_path ./pretrained_models/mosaic/add_face.pth --use_gpu 0

將視頻中的馬賽克移除,對于不同的打碼物體需要使用對應的預訓練模型進行馬賽克消除(輸出結果將儲存到 './result')

python deepmosaic.py --media_path ./result/ruoruo_add.jpg --model_path ./pretrained_models/mosaic/clean_face_HD.pth --use_gpu 0

更多的參數

使用自己的數據訓練模型

鳴謝

總結

以上是生活随笔為你收集整理的python去除视频马赛克_DeepMosaics的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。