Gunakan Python untuk menukar foto kepada avatar gaya anime.

Baru-baru ini saya menghadapi masalah, saya perlu menukar avatar saya kepada gaya anime.

▲Tukar gambar kepada gaya piksel

Tukar avatar kepada gaya anime, dengan harapan dapat mengubah foto sebenar menjadi foto sebenar sambil mengekalkan imej asal maklumat dan butiran tekstur Tukar kepada imej bukan fotorealistik gaya anime/kartun. Pada masa ini, selain API Baidu, terdapat banyak perpustakaan sumber terbuka pada Github yang boleh kami gunakan secara langsung.

Antaranya, AnimeGAN ialah penyelidikan dari Universiti Wuhan dan Universiti Teknologi Hubei Ia menggunakan gabungan pemindahan gaya neural + rangkaian musuh generatif (GAN), dan kesannya sangat konsisten dengan keperluan kita.

AnimeGAN mula-mula menggunakan rangka kerja Tensorflow, tetapi selepas menanyakan maklumat, didapati projek itu sudah menyokong rangka kerja PyTorch.

Alamat: https://github.com/bryandlee/animegan2-pytorch

Dan saya kebetulan telah melaksanakan analisis sentimen komen Weibo berdasarkan PyTorch sebelum ini . Jadi ia tidak akan menjadi beban untuk digunakan, dan anda tidak perlu memasang perpustakaan.

Pemasangan Pytorch

PyTorch ialah perpustakaan pembelajaran mesin Python sumber terbuka, berdasarkan Torch, digunakan untuk aplikasi seperti pemprosesan bahasa semula jadi. Rangka kerja pembelajaran mendalam ini boleh digunakan dalam banyak arah seperti pemodelan berangka, pemodelan imej, pemodelan teks, pemodelan audio, dsb.

Memasang Pytorch akan menjadi lebih menyusahkan daripada perpustakaan lain Jika anda pergi ke pemasangan rasmi dan memuat turun, anda perlu mendapatkan arahan pemasangan yang sesuai dengan anda mengikut konfigurasi sebenar .

Jika anda mendapati kelajuan muat turun perlahan atau anda menghadapi pelbagai masalah melalui langkah di atas, anda juga boleh mencuba tapak web berikut:

https :/ /download.pytorch.org/whl/torch_stable.html

Paparan tapak web:

Penghijrahan gaya anime

Selepas memasang Rangka kerja Pytorch, Kami kemudiannya boleh mengklon/muat turun projek animationgan2-pytorch secara setempat:

git clone https://github.com/bryandlee/animegan2-pytorch

Muat turun ke direktori tempatan di penghujung artikel seperti yang ditunjukkan di bawah:

Folder pemberat mengandungi empat pemberat Pilih berat yang sepadan untuk mencapai penghijrahan gaya animasi yang anda inginkan. Contoh gambar disimpan dalam folder input di bawah sampel, yang boleh digunakan terus untuk menguji perairan. Selain itu, saya juga mencipta folder keluaran baharu di bawah laluan yang sama untuk menyimpan imej yang diproses.

Seterusnya, kita hanya perlu menjalankan skrip test.py dalam baris arahan untuk memanggil projek Format arahan khusus adalah seperti berikut:

python test.py --checkpoint [model. laluan fail] --input_dir [direktori tempat imej input terletak] --output_dir [direktori output] --peranti [pemilihan peranti, cpu atau cuda]

Operasi sebenar:

Memandangkan ia seperti melaksanakan migrasi animasi muka, saya masing-masing menggunakan face_paint_512_v1.pt dan face_paint_512_v2.pt Kesannya adalah seperti yang ditunjukkan dalam rajah di bawah:

Kesan sebenar ①

Mari kita lihat gambar lain:

Kesan praktikal ②

Secara peribadi, saya rasa sudah selesai produk dengan berat face_paint_512_v2.pt lebih sesuai dengan citarasa saya untuk gaya anime Apa pendapat anda tentang imaginasi?

Akhirnya, saya juga mencuba paprika.pt untuk memindahkan gambar landskap kepada animasi.

▲Imej asal

Kesannya seperti yang ditunjukkan di bawah:

▲Rendering

Jika anda berminat untuk menukar avatar/gambar kepada gaya anime, anda juga boleh mencubanya~

Perhatian!

Perhatian!

Bagaimana jika selepas membaca artikel ini, anda tidak mahu memasang Pytorch, tetapi mahu mencipta muka komik anda sendiri secara langsung?

Anda boleh membuka URL ini pada komputer anda: https://huggingface.co/spaces/akhaliq/AnimeGANv2 Ini adalah APL AnimeGANv2 dalam talian Anda tidak perlu memasang sebarang rangka kerja secara setempat, dan anda boleh menukarnya secara langsung.

▲Tapak web AnimeGANv2

Atas ialah kandungan terperinci Gunakan Python untuk menukar foto kepada avatar gaya anime.. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!