DeepMosaics

DeepMosaics

英語| 中文
您可以使用它在圖像和視頻中自動刪除馬賽克，或向其添加馬賽克。
該項目基於“語義細分”和“圖像到圖像翻譯”。
在此網站上嘗試！

例子

起源	自動添加馬賽克	自動清潔馬賽克

• 與DeepCreampy相比

馬賽克圖像	deepcreampy	我們的

• 樣式轉移

起源	給梵高	到冬天

一個有趣的例子：里卡多·米洛斯（Ricardo Milos）

運行DeepMosaics

您可以通過預先構建的二進制軟件包或來源運行DeepMosaics。

在網上嘗試

您只需嘗試在此網站上刪除臉部的鑲嵌物即可。

預先建造的二進制包

對於Windows，我們將GUI版本用於易於測試。
下載此版本，以及通過[Google Drive] [百度雲，提取碼1x0A]的預訓練模型

• [幫助文件]
  
• 視頻教程=> [YouTube] [Bilibili]

注意：

• 需要Windows_x86_64，Windows10更好。
  
• 不同的預訓練模型適合不同的效果。 [介紹預訓練模型]
  
• 運行時間取決於計算機性能（GPU版本具有更好的性能，但需要安裝CUDA）。
  
• 如果無法播放輸出視頻，則可以嘗試使用Potplayer。
  
• GUI版本的更新比來源慢。
  

從源運行

先決條件

• Linux，Mac OS，Windows
• Python 3.6+
• FFMPEG 3.4.6
• Pytorch 1.0+
• CPU或NVIDIA GPU + CUDA CUDNN
  

依賴性

該代碼取決於OpenCV-Python，通過PIP安裝可用的火炬。

克隆這個倉庫

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

獲取預訓練的模型

您可以下載pre_trained模型並將其放入'./pretained_models'中。
[Google Drive] [百度雲，提取碼1x0A]
[預訓練模型簡介]

為了添加/刪除馬賽克，必須有一個模型文件mosaic_position.pth at ./pretrained_models/mosaic/mosaic_position.pth

安裝依賴項

（可選）創建虛擬環境

virtualenv mosaic
source mosaic/bin/activate

然後安裝依賴項

pip install -r requirements.txt

如果您無法構建scikit-image ，則運行export CFLAGS='-Wno-implicit-function-declaration然後嘗試重建。

簡單示例

• 添加馬賽克（輸出媒體將保存在./ result'中）
  

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

• 清潔馬賽克（輸出媒體將節省'./ result'）
  

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

如果看到錯誤Please check mosaic_position_model_path! ，檢查是否有一個名為mosaic_position.pth的模型文件./pretrained_models/mosaic/mosaic_position.pth

更多參數

如果要測試其他圖像或視頻，請參閱此文件。
[options_introduction.md]

使用您自己的數據集進行培訓

如果您想使用自己的數據集進行培訓，請參閱triending_with_your_own_own_dataset.md

致謝

該代碼從[pytorch-cyclegan and-pix2pix] [pytorch-unet] [pix2pixhd] [bisenet] [dfdnet] [gfrnet_pytorch_new] [pytorch-unet] [pytorch-unet] [pytorch-unet]。