DeepMosaics

Deepmosaics

Bahasa Inggris | 中文
Anda dapat menggunakannya untuk secara otomatis menghapus mosaik dalam gambar dan video, atau menambahkan mosaik ke dalamnya.
Proyek ini didasarkan pada "segmentasi semantik" dan "terjemahan gambar-ke-gambar".
Cobalah di situs web ini!

Contoh

asal	Auto Tambahkan Mosaic	Mosaik bersih otomatis

• Dibandingkan dengan Deepcreamppy

gambar mosaik	Deepcreamppy	milik kita

• Transfer gaya

asal	ke Van Gogh	ke musim dingin

Contoh yang menarik: Ricardo Milos ke kucing

Jalankan DeepMosaics

Anda dapat menjalankan DeepMosaics melalui paket biner yang sudah dibangun, atau dari sumber.

Cobalah di web

Anda cukup mencoba menghapus mosaik di wajah di situs web ini.

Paket Biner Pra-Built

Untuk Windows, kami menganjurkan versi GUI untuk pengujian yang mudah.
Unduh versi ini, dan model pra-terlatih melalui [Google Drive] [百度云, 提取码 1x0a]

• [Dokumen Bantuan]
  
• Tutorial Video => [YouTube] [Bilibili]

Perhatian:

• Membutuhkan windows_x86_64, windows10 lebih baik.
  
• Model pra-terlatih yang berbeda cocok untuk efek yang berbeda. [Pengantar model pra-terlatih]
  
• Waktu jalankan tergantung pada kinerja komputer (versi GPU memiliki kinerja yang lebih baik tetapi mengharuskan CUDA diinstal).
  
• Jika video output tidak dapat diputar, Anda dapat mencoba dengan Potplayer.
  
• Pembaruan versi GUI lebih lambat dari sumber.
  

Jalankan dari sumber

Prasyarat

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

Dependensi

Kode ini tergantung pada OpenCV-Python, TorchVision yang tersedia melalui PIP Install.

Klone repo ini

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

Dapatkan model pra-terlatih

Anda dapat mengunduh model pre_trained dan memasukkannya ke './pretrain_models'.
[Google Drive] [百度云, 提取码 1x0a]
[Pengantar model pra-terlatih]

Untuk menambah/menghapus mosaic, harus ada file model mosaic_position.pth at ./pretrained_models/mosaic/mosaic_position.pth

Instal dependensi

(Opsional) Buat lingkungan virtual

virtualenv mosaic
source mosaic/bin/activate

Kemudian pasang dependensi

pip install -r requirements.txt

Jika Anda tidak dapat membangun scikit-image , menjalankan export CFLAGS='-Wno-implicit-function-declaration kemudian cobalah untuk membangun kembali.

Contoh sederhana

• Tambahkan mosaic (media output akan disimpan di './result')
  

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

• Mosaic Bersih (media output akan menyimpan di './Result')
  

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

Jika Anda melihat kesalahan Please check mosaic_position_model_path! , periksa apakah ada file model bernama mosaic_position.pth di ./pretrained_models/mosaic/mosaic_position.pth

Lebih banyak parameter

Jika Anda ingin menguji gambar atau video lain, silakan merujuk ke file ini.
[option_introduction.md]

Pelatihan dengan dataset Anda sendiri

Jika Anda ingin berlatih dengan dataset Anda sendiri, silakan merujuk ke pelatihan_with_your_own_dataset.md

Ucapan Terima Kasih

Kode ini banyak meminjam dari [pytorch-cyclegan-and-pix2pix] [pytorch-unet] [pix2pixhd] [bisenet] [dfdnet] [gfrnet_pytorch_new].