DeepMosaics

DeepMosaics

Inglês | 中文
Você pode usá -lo para remover automaticamente os mosaicos em imagens e vídeos ou adicionar mosaicos a eles.
Este projeto é baseado em "segmentação semântica" e "tradução de imagem para imagem".
Experimente neste site!

Exemplos

origem	Adicione automaticamente o mosaico	Mosaico limpo automático

• Comparado com DeepCreampy

imagem em mosaico	DeepCreampy	nosso

• Transferência de estilo

origem	para Van Gogh	para o inverno

Um exemplo interessante: Ricardo Milos para Cat

Execute DeepMosaics

Você pode executar o DeepMosaics por meio de um pacote binário pré-criado ou da fonte.

Experimente na web

Você pode simplesmente tentar remover o mosaico no rosto neste site.

Pacote binário pré-criado

Para o Windows, assumimos uma versão da GUI para facilitar o teste.
Faça o download desta versão e um modelo pré-treinado via [Google Drive] [百度云, 提取码 1x0a]

• [Documento de ajuda]
  
• Tutorial em vídeo => [YouTube] [Bilibili]

Attentões:

• Requer Windows_x86_64, o Windows10 é melhor.
  
• Diferentes modelos pré-treinados são adequados para efeitos diferentes. [Introdução aos modelos pré-treinados]
  
• O tempo de execução depende do desempenho dos computadores (a versão da GPU tem melhor desempenho, mas exige que o CUDA seja instalado).
  
• Se o vídeo de saída não puder ser reproduzido, você pode tentar com o PotPlayer.
  
• A versão da GUI é atualizada mais lenta que a fonte.
  

Fugir da fonte

Pré -requisitos

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

Dependências

Este código depende do OpenCV-Python, TorchVision disponível via Pip Install.

Clone este repo

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

Obtenha modelos pré-treinados

Você pode baixar modelos pré -titred e colocá -los em './pretringen_models'.
[Google Drive] [百度云, 提取码 1x0a]
[Introdução aos modelos pré-treinados]

Para adicionar/remover o mosaico, deve haver um modelo de arquivo mosaic_position.pth em ./pretrained_models/mosaic/mosaic_position.pth

Instalar dependências

(Opcional) Crie um ambiente virtual

virtualenv mosaic
source mosaic/bin/activate

Em seguida, instale as dependências

pip install -r requirements.txt

Se você não pode criar scikit-image , a execução export CFLAGS='-Wno-implicit-function-declaration , e tente reconstruir.

Exemplo simples

• Adicione o mosaico (a mídia de saída será salva em './result')
  

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

• Mosaico limpo (a mídia de saída salvará em './result')
  

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

Se você vir o erro Please check mosaic_position_model_path! , verifique se existe um arquivo de modelo chamado mosaic_position.pth em ./pretrained_models/mosaic/mosaic_position.pth

Mais parâmetros

Se você deseja testar outras imagens ou vídeos, consulte este arquivo.
[options_introduction.md]

Treinamento com seu próprio conjunto de dados

Se você deseja treinar com seu próprio conjunto de dados, consulte Training_with_your_own_dataset.md

Agradecimentos

Este código empréstimo empréstimo fortemente de [pytorch-cyclegan-and-pix2pix] [pytorch-unet] [pix2pixhd] [bisenet] [dfdnet] [gfrnet_pytorch_new].