Commit graph

927 commits

Author SHA1 Message Date
Colombo
7e609542db fixes 2019-12-22 21:17:34 +04:00
Colombo
9ed0111824 fix 2019-12-22 19:10:57 +04:00
Colombo
754d6c385c refactorings 2019-12-22 19:00:59 +04:00
Colombo
e0e8970ab9 fix PackedFaceset 2019-12-22 15:58:46 +04:00
Colombo
dd45b7dacc cleaning 2019-12-22 15:02:53 +04:00
Colombo
8153e90ba3 refactoring 2019-12-22 14:05:41 +04:00
Colombo
246129a964 change samples.pak to faceset.pak 2019-12-22 13:32:19 +04:00
Colombo
9312a3c334 Extractor: 2nd pass now faster on frames where faces are not found 2019-12-22 10:46:44 +04:00
Colombo
8305832a27 fix 2019-12-21 23:27:51 +04:00
Colombo
50f892d57d all models: removed options 'src_scale_mod', and 'sort samples by yaw as target'
If you want, you can manually remove unnecessary angles from src faceset after sort by yaw.

Optimized sample generators (CPU workers). Now they consume less amount of RAM and work faster.

added
4.2.other) data_src/dst util faceset pack.bat
	Packs /aligned/ samples into one /aligned/samples.pak file.
	After that, all faces will be deleted.

4.2.other) data_src/dst util faceset unpack.bat
	unpacks faces from /aligned/samples.pak to /aligned/ dir.
	After that, samples.pak will be deleted.

Packed faceset load and work faster.
2019-12-21 23:16:55 +04:00
Colombo
8866dce22e added
4.2.other) data_src util faceset metadata save.bat
	saves metadata of data_src\aligned\ faces into data_src\aligned\meta.dat

4.2.other) data_src util faceset metadata restore.bat
	restore metadata from 'meta.dat' to images
	if image size different from original, then it will be automatically resized

You can greatly enhance face details of src faceset by using Topaz Gigapixel software.
example https://i.imgur.com/Gwee99L.jpg
Example of workflow:
1) run 'data_src util faceset metadata save.bat'
2) launch Topaz Gigapixel
3) open 'data_src\aligned\' and select all images
4) set output folder to 'data_src\aligned_topaz' (create folder in save dialog)
5) set settings as on screenshot https://i.imgur.com/kAVWMQG.jpg
	you can choose 2x, 4x, or 6x upscale rate
6) start process images and wait full process
7) rename folders:
	data_src\aligned        ->  data_src\aligned_original
	data_src\aligned_topaz  ->  data_src\aligned
8) copy 'data_src\aligned_original\meta.dat' to 'data_src\aligned\'
9) run 'data_src util faceset metadata restore.bat'
	images will be downscaled back to original size (256x256) preserving details
	metadata will be restored
10) now your new enhanced faceset is ready to use !
2019-12-20 15:03:17 +04:00
Colombo
5e6a1f5249 fix 2019-12-20 14:16:44 +04:00
Colombo
021bb6d128 SAEHD: lr_dropout now as an option, and disabled by default 2019-12-20 12:04:44 +04:00
Colombo
951942821d 1 2019-12-20 11:04:26 +04:00
Colombo
cf27558e88 SAEHD: lr_dropout now as an option, and disabled by default 2019-12-20 10:39:05 +04:00
Colombo
64021b9c62 more stable and precise version of face transformation matrix.
fixed bleeding mask on some samples
2019-12-20 10:30:49 +04:00
Colombo
068c7d0d55 temporary revert last fixes 2019-12-20 10:21:59 +04:00
Colombo
dd1d5e8909 improved face align,
More stable and precise version of the face transformation matrix.
Now full_faces are aligned with the upper and lateral boundaries of the frame,
result: fix of cutted mouth, increase area of the cheeks of side faces
before/after https://i.imgur.com/t9IyGZv.jpg
therefore, additional training is required for existing models.
Optionally, you can re-extract dst faces of your project, if they have problems with cutted mouth or cheeks.
2019-12-19 18:33:04 +04:00
Colombo
9e9dc364c9 temporary revert fix 2019-12-19 15:46:50 +04:00
Colombo
853a056769 more stable and precise version of face transformation matrix 2019-12-19 15:25:06 +04:00
Colombo
c04740eac1 1 2019-12-17 09:53:17 +04:00
Colombo
28b948888c fix extractor for some systems 2019-12-17 09:52:48 +04:00
Colombo
71ebf06c89 SAEHD,Quick96:
improved model generalization, overall accuracy and sharpness
by using new 'Learning rate dropout' technique from paper https://arxiv.org/abs/1912.00144
An example of a loss histogram where this function is enabled after the red arrow:
https://i.imgur.com/3olskOd.jpg
2019-12-15 15:53:06 +04:00
Colombo
c866448645 clean 2019-12-11 22:34:22 +04:00
Colombo
d4745b5cf8 added sort by absdiff
This is sort method by absolute per pixel difference between all faces.
options:
Sort by similar? ( y/n ?:help skip:y ) :
if you choose 'n', then most dissimilar faces will be placed first.
2019-12-11 22:33:49 +04:00
Colombo
e8673e3fcc nothing interesting 2019-12-11 22:33:23 +04:00
Colombo
154820a954 ConvertMasked: fix color bug for superresolution 2019-12-06 16:49:27 +04:00
Colombo
f7920d3880 1 2019-11-24 20:33:34 +04:00
Colombo
77b390c04b 1 2019-11-24 19:51:07 +04:00
Colombo
1bfd65abe5 extractor for AMD gpu: increase no response time 2019-11-24 19:37:21 +04:00
Colombo
6b14741160 1 2019-11-21 11:40:55 +04:00
Colombo
668513a507 1 2019-11-16 16:54:10 +04:00
Colombo
b74c2b1f9a Converter: added new color transfer mode: mix-m 2019-11-14 12:08:27 +04:00
Colombo
1cc24f2a75 1 2019-11-13 10:31:04 +04:00
Colombo
0a1a9b4fed fix 2019-11-13 00:29:41 +04:00
Colombo
006233656b facerelighther: added intensity param 2019-11-13 00:19:26 +04:00
Colombo
54786848bc fix converterconfig 2019-11-13 00:19:01 +04:00
Colombo
8c3fa3baa5 fix 2019-11-12 18:02:23 +04:00
Colombo
2167c5cbcb fix of converter when using relighted faces in dst 2019-11-12 17:37:13 +04:00
Colombo
4868bd1324 upd doc/doc_prebuilt_windows_app.md 2019-11-12 09:16:42 +04:00
Colombo
c0f258c336 SAE,SAEHD,Converter:
added sot-m color transfer

Converter:
removed seamless2 mode
2019-11-12 09:07:50 +04:00
Colombo
05153d9ba5 FacesetRelighter fixes and improvements:
now you have 3 ways:
1) define light directions manually (not for google colab)
   watch demo https://youtu.be/79xz7yEO5Jw
2) relight faceset with one random direction
3) relight faceset with predefined 8 directions
2019-11-11 19:56:36 +04:00
Colombo
fe58459f36 added FacesetRelighter:
Synthesize new faces from existing ones by relighting them using DeepPortraitRelighter network.
With the relighted faces neural network will better reproduce face shadows.

Therefore you can synthsize shadowed faces from fully lit faceset.
https://i.imgur.com/wxcmQoi.jpg

as a result, better fakes on dark faces:
https://i.imgur.com/5xXIbz5.jpg

in OpenCL build Relighter runs on CPU,

install pytorch directly via pip install, look at requirements
2019-11-11 11:42:52 +04:00
Colombo
b9c0815d17 added random shuffle to samples in SampleGeneratorFace.py 2019-11-11 11:35:46 +04:00
Colombo
f6f8f8ee5d change default jpg face extractor quality to 100 2019-11-11 09:13:42 +04:00
Colombo
7d389718fe added Quick96 model
This is the fastest model for low-end cards.
Model has zero options and trains a 96pix fullface.
It is good for quick deepfake demo.
Example of the preview trained in 15 minutes on RTX2080Ti:
https://i.imgur.com/oRMvZFP.jpg
2019-11-09 19:24:31 +04:00
Colombo
1f350ae413 1 2019-11-09 19:23:55 +04:00
Colombo
770c70d778 converter:
fixed crashes

removed useless 'ebs' color transfer

changed keys for color degrade

added image degrade via denoise - same as denoise extracted data_dst.bat ,
but you can control this option directly in the interactive converter

added image degrade via bicubic downscale and upscale

SAEHD: default ae_dims for df now 256.
2019-11-09 15:12:35 +04:00
Colombo
374d8c2388 fix denoise_image_sequence 2019-11-08 23:04:51 +04:00
Colombo
d897893fd5 fix denoise-image-sequence for jpg files 2019-11-08 22:58:39 +04:00