Commit Graph

701 Commits

Author SHA1 Message Date
Colombo 2429d28737 optimize memory usage 2020-01-04 23:51:33 +04:00
Colombo f95682f616 fix 2020-01-02 21:19:19 +04:00
Colombo 5efb430f2a +WScaleConv2DLayer 2020-01-01 11:15:55 +04:00
Colombo efd8c16daa avatar: avatar_type is now only head by default 2019-12-29 19:37:59 +04:00
Colombo 47e539ccdd fix extract unaligned faces 2019-12-29 19:36:34 +04:00
Colombo 818a2d65f0 fix long load when using various gpus in the same DFL folder 2019-12-29 18:47:55 +04:00
Colombo 540650c610 fix faceset enhancer for frames that contain edited mask 2019-12-29 18:10:05 +04:00
Colombo 28549dc153 SAEHD:optimized architecture, you have to restart training 2019-12-28 16:50:33 +04:00
Colombo 4e67278c66 fix 0 faces detected in manual extractor 2019-12-28 16:45:40 +04:00
Colombo 3849172a43 add GeLU 2019-12-27 21:25:59 +04:00
Colombo 0833a38bb9 Quick96: optimized architecture 2019-12-27 21:06:45 +04:00
Colombo a347bb4f15 change doc about builds 2019-12-27 14:29:27 +04:00
Colombo 7804e0bd7c add try except 2019-12-27 13:36:01 +04:00
Colombo df8cf35c75 FacesetEnhancer now asks to merge aligned_enhanced/ to aligned/ 2019-12-27 11:09:15 +04:00
Colombo d46fb5cfd3 fixed mask editor
added FacesetEnhancer
4.2.other) data_src util faceset enhance best GPU.bat
4.2.other) data_src util faceset enhance multi GPU.bat

FacesetEnhancer greatly increases details in your source face set,
same as Gigapixel enhancer, but in fully automatic mode.
In OpenCL build it works on CPU only.

Please consider a donation.
2019-12-26 21:27:10 +04:00
Colombo 3be223a265 fix MaskEditor 2019-12-25 11:13:52 +04:00
Colombo ac1913c070 fix 2019-12-24 14:07:06 +04:00
Colombo 3feb386a8e fix for colab 2019-12-24 11:20:32 +04:00
Colombo c482b2cd07 upd reaedme 2019-12-23 19:16:33 +04:00
Colombo b93ae56a08 upd readme 2019-12-23 19:15:59 +04:00
Colombo 8c17dd85f4 upd readme 2019-12-23 19:06:30 +04:00
Colombo 94c99b429d fix 2019-12-23 15:17:13 +04:00
Colombo c1612c5553 fixes 2019-12-23 14:57:47 +04:00
Colombo 7174dc835a fixes 2019-12-22 21:44:28 +04:00
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