Commit Graph
100 Commits
Author SHA1 Message Date
Colombo d78ec338c6 fix Subprocessor now writes an error if it fails to process the data 2020-06-24 11:44:58 +04:00
Colombo ddb6bcf416 fix 2020-06-24 11:44:24 +04:00
Colombo f0508287f0 added de-age the face example 2020-06-24 10:15:44 +04:00
Colombo 4e998aff93 . 2020-06-24 10:06:31 +04:00
Colombo 29b1050637 Sorter: improved sort by blur, and sort by best faces 2020-06-23 21:02:38 +04:00
Colombo 669935c9e6 XSegEditor: changed hotkey for xseg overlay mask 2020-06-22 13:13:27 +04:00
Colombo d0ded46b51 XSegEditor: overlay xseg mask now works in polygon mode https://i.imgur.com/wDV6PY2.jpg 2020-06-22 11:12:24 +04:00
Colombo f90723c9f9 SAEHD: ‘uniform_yaw’ now always enabled in pretrain mode. 2020-06-21 00:57:48 +04:00
Colombo 27f219fec7 XSegEditor: fixed incorrect count of labeled images. 2020-06-21 00:56:25 +04:00
Colombo 2b8e8f0554 . 2020-06-20 22:05:30 +04:00
Colombo 53222839c6 SAEHD: for -d archis resolution should be divided by 32 2020-06-19 14:21:41 +04:00
Colombo 05606b5e15 Subprocessor: now should write an error if failed to start 2020-06-19 14:20:22 +04:00
Colombo 0c2e1c3944 SAEHD:
Maximum resolution is increased to 640.

‘hd’ archi is removed. ‘hd’ was experimental archi created to remove subpixel shake, but ‘lr_dropout’ and ‘disable random warping’ do that better.

‘uhd’ is renamed to ‘-u’
dfuhd and liaeuhd will be automatically renamed to df-u and liae-u in existing models.

Added new experimental archi (key -d) which doubles the resolution using the same computation cost.
It is mean same configs will be x2 faster, or for example you can set 448 resolution and it will train as 224.
Strongly recommended not to train from scratch and use pretrained models.

New archi naming:
'df' keeps more identity-preserved face.
'liae' can fix overly different face shapes.
'-u' increased likeness of the face.
'-d' (experimental) doubling the resolution using the same computation cost
Examples: df, liae, df-d, df-ud, liae-ud, ...

Improved GAN training (GAN_power option).  It was used for dst model, but actually we don’t need it for dst.
Instead, a second src GAN model with x2 smaller patch size was added, so the overall quality for hi-res models should be higher.

Added option ‘Uniform yaw distribution of samples (y/n)’:
	Helps to fix blurry side faces due to small amount of them in the faceset.

Quick96:
	Now based on df-ud archi and 20% faster.

XSeg trainer:
	Improved sample generator.
Now it randomly adds the background from other samples.
Result is reduced chance of random mask noise on the area outside the face.
Now you can specify ‘batch_size’ in range 2-16.

Reduced size of samples with applied XSeg mask. Thus size of packed samples with applied xseg mask is also reduced.
2020-06-19 09:45:55 +04:00
Colombo 9fd3a9ff8d add meme 2020-06-18 22:15:25 +04:00
Colombo 38573473da added youtube channel 2020-06-18 14:19:58 +04:00
Colombo 9994f1512e added third head swap example 2020-06-18 13:56:42 +04:00
Colombo 11327cb0c5 upd face examples 2020-06-17 10:51:11 +04:00
Colombo 70182f8c65 upd face examples 2020-06-17 10:49:41 +04:00
Colombo 8c467bad90 fuck linux 2020-06-15 11:26:22 +04:00
Colombo 2e650a7122 . 2020-06-12 10:45:57 +04:00
Colombo 7f11713730 fix 2020-06-11 21:29:39 +04:00
Colombo d99afebbdc new telegram group link 2020-06-11 17:39:36 +04:00
Colombo 82f405ed49 Trainer: fixed "Choose image for the preview history". Now you can switch between subpreviews using 'space' key.
Fixed "Write preview history". Now it writes all subpreviews in separated folders
https://i.imgur.com/IszifCJ.jpg
also the last preview saved as _last.jpg before the first file
https://i.imgur.com/Ls1AOK4.jpg
thus you can easily check the changes with the first file in photo viewer
2020-06-10 10:20:13 +04:00
Colombo 5c315cab68 XSegEditor: added text label of total labeled images
Changed frame line design
Changed loading frame design
2020-06-09 22:04:15 +04:00
Colombo dc43f5a891 add second headswap example 2020-06-09 16:01:47 +04:00
Colombo c2b5f80e3f upd readme download link 2020-06-08 15:03:40 +04:00
Colombo 1d3d417aee upd readme 2020-06-08 14:53:19 +04:00
Colombo af98407f06 SAEHD: lr_dropout now can be ‘n’, ‘y’, ‘cpu’. ‘n’ and ’y’ are the same as before.
‘cpu’ mean enabled on CPU. This allows not to use extra VRAM, sacrificing 20% time of iteration.

SAEHD: resolution >= 256 now has second dssim loss function
2020-06-08 14:18:33 +04:00
Colombo 4acaf08ebf fix 2020-06-08 14:07:47 +04:00
Colombo cfd7803e0d leras : DepthwiseConv2D 2020-06-07 20:59:12 +04:00
Colombo 85ede1a7e7 . 2020-06-07 12:26:23 +04:00
Colombo ddf7363eda remove unused code 2020-06-06 20:02:21 +04:00
Colombo 235315dd70 fix 2020-06-06 19:52:07 +04:00
Colombo 1e003170bd reduced chance of "The paging file is too small for this operation to complete." 2020-06-06 19:38:11 +04:00
Colombo 61f93063b2 fix 2020-06-04 16:44:29 +04:00
Colombo addc96fe3e fix 2020-06-04 16:42:36 +04:00
Colombo 65bc9273e6 upd readme 2020-06-04 15:38:20 +04:00
Colombo c94c9106be upd readme 2020-06-04 14:44:47 +04:00
Colombo 903657ff91 upd readme 2020-06-04 14:39:50 +04:00
Colombo f91a604c6d Manual extractor: now you can specify face rectangle manually using ‘R Mouse button’.
It is useful for small, blurry, undetectable faces, animal faces.
https://i.imgur.com/8kmVgg8.jpg
Warning: such frames must be used only with XSeg workflow !
Landmarks cannot be placed on the face precisely, and they are actually used for positioning the red frame.
Try to keep the red frame the same as the adjacent frames.
2020-06-02 12:31:24 +04:00
Colombo e77865ce18 XSegEditor: fix 2020-06-01 13:44:29 +04:00
Colombo 6f25ebbc9e XSegEditor: added button "view XSeg mask overlay face" 2020-05-31 18:00:39 +04:00
Colombo c6652ac006 add arxiv paper 2020-05-24 15:10:40 +04:00
Colombo 130d72cb8b upd 2020-05-22 22:44:00 +04:00
Colombo 72e2f771a2 upd readme 2020-05-20 09:58:55 +04:00
Colombo 0c18b1f011 changed uhd arhi: you have to retrain uhd arhis from scratch. 2020-05-06 19:21:23 +04:00
Colombo dcf380345b fix 2020-05-02 14:04:32 +04:00
Colombo f935bf0465 . 2020-04-28 19:04:44 +04:00
Colombo 7b49220ee9 fix 2020-04-20 18:40:12 +04:00
Colombo 4fe45adc7c . 2020-04-20 10:24:51 +04:00
Colombo 3a9e851339 fix bug 2020-04-18 18:29:55 +04:00
Colombo a3c91271d0 upd README.md : google drive is replaced by mega 2020-04-16 00:08:42 +04:00
Colombo baff59e421 . 2020-04-15 22:58:59 +04:00
Colombo ea607edfc9 XSegEditor: added view lock at the center by holding shift. 2020-04-15 22:12:54 +04:00
Colombo 6bbc607312 color transfer sot : optimization 2020-04-15 18:13:03 +04:00
Colombo f8580928ed SampleLoader: fix error 2020-04-15 16:55:51 +04:00
Colombo c42b1b2124 . 2020-04-15 10:44:12 +04:00
Colombo cd8919d95b Merger: optimizations 2020-04-14 10:25:54 +04:00
Colombo e549624eeb upd README : remove 'github releases' 2020-04-13 22:33:17 +04:00
Colombo 2cc0e64572 Merger: added mask mode ‘learned-prd + learned-dst’ – produces largest area of both dst and predicted masks 2020-04-13 22:04:24 +04:00
Colombo d1af3b51cd . 2020-04-13 21:37:55 +04:00
Colombo 835f3adb8e XSegEditor : polygon is now transparent while editing 2020-04-13 21:37:45 +04:00
Colombo 4a3b94832f upd 2020-04-13 21:02:51 +04:00
Colombo 95847637e5 upd 2020-04-13 21:02:03 +04:00
Colombo 8ddc7954da upd README.md : added mini tutorial 2020-04-13 20:57:51 +04:00
Colombo 80d4eef1fe . 2020-04-11 23:38:35 +04:00
Colombo 23961879b4 Merger: color transfer ‘sot-m’ : reduced color flickering consuming x5 more time to process 2020-04-11 23:33:55 +04:00
Colombo 25324267d6 Merger: color transfer ‘sot-m’ : reduced color flickering 2020-04-11 21:33:47 +04:00
Colombo f72df02950 upd 2020-04-08 18:20:57 +04:00
Colombo 935d940ace fix 2020-04-07 22:06:32 +04:00
Colombo d8a792232c upd readme 2020-04-07 10:25:11 +04:00
Colombo 5b4d023712 added 5.XSeg) data_dst/data_src mask for XSeg trainer - remove.bat
removes labeled xseg polygons from the extracted frames
2020-04-06 21:45:46 +04:00
Colombo b6b3936bcd update readme 2020-04-06 20:15:46 +04:00
Colombo f821ab350f fix 2020-04-06 19:06:37 +04:00
Colombo afdb1ef85d fix for 16+ CPU 2020-04-06 18:42:38 +04:00
Colombo 453237bfd7 upd README.md 2020-04-06 17:55:33 +04:00
Colombo 215277b376 XSegEditor: fix 2020-04-05 14:14:56 +04:00
Colombo 33b0aadb4e Decreased amount of RAM used by Sample Generator. 2020-04-05 13:52:32 +04:00
Colombo 8e9e346c9d Extractor: now face_type can be choosed in console dialog 2020-04-04 10:08:23 +04:00
Colombo 8aa87c3080 upd README.md 2020-04-04 09:41:10 +04:00
Colombo 4db11aa133 upd README.md 2020-04-04 09:40:18 +04:00
Colombo 2b7364005d Added new face type : head
Now you can replace the head.
Example: https://www.youtube.com/watch?v=xr5FHd0AdlQ
Requirements:
	Post processing skill in Adobe After Effects or Davinci Resolve.
Usage:
1)	Find suitable dst footage with the monotonous background behind head
2)	Use “extract head” script
3)	Gather rich src headset from only one scene (same color and haircut)
4)	Mask whole head for src and dst using XSeg editor
5)	Train XSeg
6)	Apply trained XSeg mask for src and dst headsets
7)	Train SAEHD using ‘head’ face_type as regular deepfake model with DF archi. You can use pretrained model for head. Minimum recommended resolution for head is 224.
8)	Extract multiple tracks, using Merger:
a.	Raw-rgb
b.	XSeg-prd mask
c.	XSeg-dst mask
9)	Using AAE or DavinciResolve, do:
a.	Hide source head using XSeg-prd mask: content-aware-fill, clone-stamp, background retraction, or other technique
b.	Overlay new head using XSeg-dst mask

Warning: Head faceset can be used for whole_face or less types of training only with XSeg masking.

XSegEditor: added button ‘view trained XSeg mask’, so you can see which frames should be masked to improve mask quality.
2020-04-04 09:28:06 +04:00
Colombo 383d4d3736 fix 2020-04-03 16:24:34 +04:00
Colombo 0fb912e91f Trainer: added --silent-start cmd option 2020-04-02 13:05:04 +04:00
Colombo 2fe86faf01 fix Win10 input dialog 2020-04-02 10:27:39 +04:00
Colombo dbb0988927 upd README.md 2020-04-01 22:17:59 +04:00
Colombo 652ef75099 upd README.md 2020-04-01 16:08:06 +04:00
Colombo 3702531898 SAEHD: ‘Face style power’ and ‘Background style power’ are now available for whole_face
New help messages for these options:

Face style power
Learn the color of the predicted face to be the same as dst inside mask.
If you want to use this option with 'whole_face' you have to use XSeg trained mask.
Warning: Enable it only after 10k iters, when predicted face is clear enough to start learn style.
Start from 0.001 value and check history changes.
Enabling this option increases the chance of model collapse

Background style power
      Learn the area outside mask of the predicted face to be the same as dst.
If you want to use this option with 'whole_face' you have to use XSeg trained mask.
This can make face more like dst.
Enabling this option increases the chance of model collapse. Typical value is 2.0
2020-03-30 19:46:17 +04:00
Colombo 9a9b7e4f81 fix 2020-03-30 14:39:07 +04:00
Colombo 6d3607a13d New script:
5.XSeg) data_dst/src mask for XSeg trainer - fetch.bat
Copies faces containing XSeg polygons to aligned_xseg\ dir.
Useful only if you want to collect labeled faces and reuse them in other fakes.

Now you can use trained XSeg mask in the SAEHD training process.
It’s mean default ‘full_face’ mask obtained from landmarks will be replaced with the mask obtained from the trained XSeg model.
use
5.XSeg.optional) trained mask for data_dst/data_src - apply.bat
5.XSeg.optional) trained mask for data_dst/data_src - remove.bat

Normally you don’t need it. You can use it, if you want to use ‘face_style’ and ‘bg_style’ with obstructions.

XSeg trainer : now you can choose type of face
XSeg trainer : now you can restart training in “override settings”
Merger: XSeg-* modes now can be used with all types of faces.

Therefore old MaskEditor, FANSEG models, and FAN-x modes have been removed,
because the new XSeg solution is better, simpler and more convenient, which costs only 1 hour of manual masking for regular deepfake.
2020-03-30 14:00:40 +04:00
Colombo e5bad483ca upd readme 2020-03-30 09:53:46 +04:00
Colombo 3d4b9c858e deepfaker.xyz - 非官方网站帮助中国人学习DeepFaceLab 2020-03-29 21:48:47 +04:00
Colombo 497a7eec94 fix preview_history 2020-03-29 14:50:02 +04:00
Colombo 4e744cf184 Colab: change save preview iters to every 100 2020-03-28 13:58:41 +04:00
Colombo c3ce06a588 facetype.mouth 2020-03-28 09:38:09 +04:00
Colombo d1a5639e90 XSegEditor: fix bugs 2020-03-27 19:08:01 +04:00
Colombo 6687213fa5 SAEHD: fixed the use of pretrained liae model, now it produces less face morphing 2020-03-27 00:15:18 +04:00
Colombo ca9138d6b7 XSegEditor: changed layout, added current filename 2020-03-27 00:11:36 +04:00
Colombo 8ad2a5373e fix Exception("nn devices are not initialized. Run initialize_main_env() in main process.") 2020-03-24 22:06:10 +04:00
Colombo 2d5e949100 fix Chinesse language 2020-03-24 19:31:17 +04:00