Commit Graph

52 Commits

Author SHA1 Message Date
iperov b21bce458f Sorter: added sort by motion blur 2020-12-13 09:49:32 +04:00
Colombo b6dd482e05 . 2020-06-27 19:08:14 +04:00
Colombo 29b1050637 Sorter: improved sort by blur, and sort by best faces 2020-06-23 21:02:38 +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 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 efe3b56683 DFLIMG refactoring 2020-03-21 01:18:15 +04:00
Colombo 61472cdaf7 global refactoring and fixes,
removed support of extracted(aligned) PNG faces. Use old builds to convert from PNG to JPG.

fanseg model file in facelib/ is renamed
2020-03-13 08:09:00 +04:00
Colombo 9ccdd271a4 added sort by "best faces faster"
same as sort by "best faces"
but faces will be sorted by source-rect-area instead of blur, x40 times faster.
2020-03-06 21:38:41 +04:00
Colombo f56d583cb5 added sort by "face rect size in source image"
small faces from source image will be placed at the end
2020-03-03 21:47:49 +04:00
Colombo d65ff888cd fix sort by best. 2020-02-24 20:52:02 +04:00
Colombo 4ea70864fc fix Sorter.py 2020-02-23 20:19:13 +04:00
Colombo 75a1adc7c4 fix sort by yaw/pitch for whole_face 2020-02-22 13:43:50 +04:00
Colombo 76ca79216e Upgraded to TF version 1.13.2
Removed the wait at first launch for most graphics cards.

Increased speed of training by 10-20%, but you have to retrain all models from scratch.

SAEHD:

added option 'use float16'
	Experimental option. Reduces the model size by half.
	Increases the speed of training.
	Decreases the accuracy of the model.
	The model may collapse or not train.
	Model may not learn the mask in large resolutions.

true_face_training option is replaced by
"True face power". 0.0000 .. 1.0
Experimental option. Discriminates the result face to be more like the src face. Higher value - stronger discrimination.
Comparison - https://i.imgur.com/czScS9q.png
2020-01-25 21:58:19 +04:00
Colombo 38b85108b3 DFL-2.0 initial branch commit 2020-01-21 18:43:39 +04:00
Colombo 7174dc835a fixes 2019-12-22 21:44:28 +04:00
Colombo 8153e90ba3 refactoring 2019-12-22 14:05:41 +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 734d97d729 added 'sort by vggface': sorting by face similarity using VGGFace model.
Requires 4GB+ VRAM and internet connection for the first run.
2019-10-23 15:06:39 +04:00
Colombo d3e35c8ce6 enhanced sort by final: now faces are evenly distributed not only in the direction of yaw, but also in pitch 2019-10-20 14:18:14 +04:00
iperov b72d5a3f9a fixed error "Failed to get convolution algorithm" on some systems
fixed error "dll load failed" on some systems
Expanded eyebrows line of face masks. It does not affect mask of FAN-x converter mode.
2019-08-11 11:17:22 +04:00
iperov e58197ca22 initial code to extract umdfaces.io dataset and train pose estimator 2019-04-23 08:14:09 +04:00
iperov b03b147bae refactoring 2019-03-26 11:09:44 +04:00
iperov a3df04999c removing trailing spaces 2019-03-19 23:53:27 +04:00
iperov a0c1ce3e7e final sort: increased speed 2019-03-19 13:45:26 +04:00
iperov 7c66503382 sorter: fix final sort 2019-03-16 09:20:55 +04:00
iperov ea5fe28d08 fix final sorter 2019-03-15 23:41:34 +04:00
iperov 3549272244 fix for final 2019-03-15 11:03:37 +04:00
iperov 5bd3f875fa fix 2019-03-14 10:50:53 +04:00
iperov 6e59f1a150 Sorter: new better blur estimator 2019-03-14 10:38:56 +04:00
iperov 438213e97c manual extractor: increased FPS,
sort by final : now you can specify target number of images,
converter: fix seamless mask and exception,
huge refactoring
2019-02-28 11:56:31 +04:00
iperov 5bbd8997d9 fix Sorter.py 2019-02-22 15:22:42 +04:00
iperov 5e474c3462 added sort by one face in images. it trashes all images in which more than 1 face is found 2019-02-22 11:32:27 +04:00
iperov 06fe1314d8 removing default yaw_value from DFLIMG files,
added better pitch/yaw estimator from 68 landmarks,
improving face yaw accuracy for sorting and trainers,
added sort by face-pitch
2019-02-12 21:31:37 +04:00
iperov 5a1fb199f4 added support of non-english characters in paths 2019-02-05 19:33:55 +04:00
iperov 6d95dd4a99 DFL now works with JPG files by default. Old PNG files also supported. Added util convertor PNG to JPG. 2019-02-04 19:47:24 +04:00
iperov 72646becd1 forcing all tqdm's to ascii in order to work properly on Chinese windows 2019-01-24 21:32:07 +04:00
iperov b6c4171ea1 optimizations of nnlib and SampleGeneratorFace,
refactorings
2019-01-22 11:52:04 +04:00
iperov cf41e1b962 fix sort by final 2019-01-07 17:27:10 +04:00
iperov ff3a208e80 fix sort by final 2019-01-07 14:57:46 +04:00
iperov abdbe7fd8d added sort by 'final' 2019-01-07 14:20:44 +04:00
iperov b5ba7d52cb increased speed of sort by hist sim for ten thousands of faces 2019-01-04 22:54:09 +04:00
iperov 57e1e59c03 revert sort_by_hist 2019-01-02 23:53:27 +04:00
iperov a37af3e965 remove print_on_no_embedded_data=True on sort_by_hist_dissim 2018-12-25 19:50:25 +04:00
iperov d04e8b1d91 increased speed of sort by hist 2018-12-20 18:04:13 +04:00
iperov bc5ca1ab59 removed sort by hist-blur 2018-12-20 12:57:16 +04:00
iperov 9926dc626a refactorings, improved sort by hist-dissim 2018-12-20 12:43:00 +04:00
iperov 26571eff6a added sort -by black . Places images which contains black area at end of folder. Useful to get rid of src faces which cutted by screen. 2018-12-18 18:08:04 +04:00
iperov db142bcc02 increased speed of 'sort by hist-dissim' by running in parallel 2018-12-18 17:51:54 +04:00