Téléchargement DeepMosaics - Téléchargement du code source DeepMosaics

DeepMosaics

Python

v0.5.1

Télécharger

Deepmosaïque

Anglais | 中文
Vous pouvez l'utiliser pour supprimer automatiquement les mosaïques en images et vidéos, ou y ajouter des mosaïques.
Ce projet est basé sur la "segmentation sémantique" et la "traduction d'image à l'image".
Essayez-le sur ce site!

Exemples

origine	Ajouter automatiquement la mosaïque	mosaïque propre

Par rapport à Deepcreampy

image mosaïque	Profondément	la nôtre

Transfert de style

origine	à Van Gogh	à l'hiver

Un exemple intéressant: Ricardo Milos à Cat

Exécuter Deepmosaics

Vous pouvez soit exécuter DeepMosaics via un package binaire prédéfini, soit à partir de Source.

Essayez-le sur le Web

Vous pouvez simplement essayer de supprimer la mosaïque sur le visage sur ce site.

Ensemble binaire pré-construit

Pour Windows, nous avons gonflé une version GUI pour des tests faciles.
Téléchargez cette version et un modèle pré-formé via [Google Drive] [百度云, 提取码 1x0a]

[Aide à documenter]
Tutorial vidéo => [YouTube] [Bilibili]

Attentions:

Nécessite Windows_x86_64, Windows10 est meilleur.
Différents modèles pré-formés conviennent à différents effets. [Introduction aux modèles pré-formés]
Le temps d'exécution dépend des performances des ordinateurs (la version GPU a de meilleures performances mais nécessite l'installation de CUDA).
Si la vidéo de sortie ne peut pas être lue, vous pouvez essayer avec PotPlayer.
La version GUI met à jour plus lentement que la source.

Courir à partir de la source

Condition préalable

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

Dépendances

Ce code dépend d'OpenCV-Python, TorchVision disponible via PIP Installer.

Cloner ce repo

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

Obtenez des modèles pré-formés

Vous pouvez télécharger des modèles pré_trained et les mettre dans './pretrained_models'.
[Google Drive] [百度云, 提取码 1x0a]
[Introduction aux modèles pré-formés]

Afin d'ajouter / supprimer la mosaïque, il doit y avoir un fichier de modèle mosaic_position.pth à ./pretrained_models/mosaic/mosaic_position.pth

Installer des dépendances

(Facultatif) Créer un environnement virtuel

virtualenv mosaic
source mosaic/bin/activate

Puis installez les dépendances

pip install -r requirements.txt

Si vous ne pouvez pas construire scikit-image , exécuter export CFLAGS='-Wno-implicit-function-declaration , essayez de reconstruire.

Exemple simple

Ajouter la mosaïque (les supports de sortie seront enregistrés dans './result')

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

Clean Mosaic (les supports de sortie seront économisés dans «./result»)

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

Si vous voyez l'erreur Please check mosaic_position_model_path! , Vérifiez s'il existe un fichier modèle nommé mosaic_position.pth à ./pretrained_models/mosaic/mosaic_position.pth