Commit Graph

42 Commits

Author SHA1 Message Date
iperov
91f0f30a78 fix save in trainer 2021-10-11 15:02:41 +04:00
iperov
5783191849 AMP: code refactoring, fix preview history
added dumpdflive command
2021-06-26 10:44:41 +04:00
iperov
1981ed0ca8 _ 2021-06-01 00:21:27 +04:00
iperov
757283d10e Trainer now saves every 25 min instead of 15 2021-05-25 14:23:25 +04:00
iperov
b333fcea4b dump_ckpt 2021-03-23 15:00:24 +04:00
iperov
554217d026 added "Windows 10 users IMPORTANT notice" on training start 2020-12-20 16:55:25 +04:00
Colombo
383d4d3736 fix 2020-04-03 16:24:34 +04:00
Colombo
9365e42f25 trainer now creates directories instead of throwing error when they do not exist 2020-03-04 17:54:52 +04:00
Colombo
a5783df546 fix autobackup_hour 2020-02-27 11:13:45 +04:00
Colombo
3e63ba2d17 fix mean loss history after save 2020-01-30 07:49:45 +04:00
Colombo
0251eb3490 MultiGPU training:
speed is significantly increased.
fixed CUDNN_STREAM errors.

Trainer: added key 'b' : creates a backup even if the autobackup is disabled.
2020-01-29 10:55:51 +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
b5c234dac3 no_preview for model 2020-01-04 23:52:12 +04:00
Colombo
dc11ec32be SAE : WARNING, RETRAIN IS REQUIRED !
fixed model sizes from previous update.
avoided bug in ML framework(keras) that forces to train the model on random noise.

Converter: added blur on the same keys as sharpness

Added new model 'TrueFace'. This is a GAN model ported from https://github.com/NVlabs/FUNIT
Model produces near zero morphing and high detail face.
Model has higher failure rate than other models.
Keep src and dst faceset in same lighting conditions.
2019-09-19 11:13:56 +04:00
Luke Barr
a33ef50da5 Long startup time on Linux (#356) 2019-08-19 22:56:54 +04:00
iperov
35ab63f752 add launch prog by timer repeatedly for colab 2019-05-05 17:05:03 +04:00
iperov
54fa75aa13 fix 2019-05-02 00:39:18 +04:00
iperov
2a8dd788dc SAE: added option 'Pretrain the model?',
Pretrain the model with large amount of various faces. This technique may help to train the fake with overly different face shapes and light conditions of src/dst data. Face will be look more like a morphed. To reduce the morph effect, some model files will be initialized but not be updated after pretrain: LIAE: inter_AB.h5 DF: both decoders.h5. The longer you pretrain the model the more morphed face will look. After that, save and run the training again.
2019-05-01 19:55:27 +04:00
iperov
cfae9f9fd8 trainer: fix for ctrl-c in --no-preview mode 2019-04-05 21:50:46 +04:00
iperov
de8d75b4f7 fix mask editor,
upd manuals
2019-04-04 22:19:56 +04:00
iperov
5ac7e5d7f1 changed help message for pixel loss:
Pixel loss may help to enhance fine details and stabilize face color. Use it only if quality does not improve over time.

SAE:
previous SAE model will not work with this update.
Greatly decreased chance of model collapse.
Increased model accuracy.
Residual blocks now default and this option has been removed.
Improved 'learn mask'.
Added masked preview (switch by space key)

Converter:
fixed rct/lct in seamless mode
added mask mode (6) learned*FAN-prd*FAN-dst

added mask editor, its created for refining dataset for FANSeg model, and not for production, but you can spend your time and test it in regular fakes with face obstructions
2019-04-04 10:22:53 +04:00
iperov
fedeee4395 Colab: added ability to run programs after N secs in trainer 2019-03-26 19:06:19 +04:00
iperov
96de328221
upd for Colab (#206)
* changes for google colab

* 1

* 1

* 1

* 1
2019-03-26 16:09:22 +04:00
iperov
b03b147bae refactoring 2019-03-26 11:09:44 +04:00
iperov
131b2b5c79 after save loss string now shows averaged value since last save 2019-03-25 18:14:04 +04:00
iperov
a3df04999c removing trailing spaces 2019-03-19 23:53:27 +04:00
iperov
97b6fabaab change 'epoch' to 'iter',
added timestamp prefix to training string
2019-03-12 19:23:52 +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
f0a20b46d3 SAE: added new archi 'vg' 2019-02-21 17:53:59 +04:00
andy-ger
cd4e5b9ba6 Fixed typo (#156) 2019-02-20 18:12:30 +04:00
iperov
72c7db4980 trainer preview: added [l]: change history range 2019-02-15 19:27:44 +04:00
iperov
a65b557b0e loss history with epoch now saves to preview history images 2019-02-07 22:34:51 +04:00
iperov
0d76bee2ee fix Trainer.py for debug 2019-02-07 11:37:55 +04:00
iperov
946688567d enhanced SAE model. You should to restart training.
new default style power = 2.0
fix DF default batch sizes.
upd readme
2019-01-17 21:41:40 +04:00
iperov
2473dabf37 fix loss history 2019-01-11 19:19:29 +04:00
iperov
7b70e7eec1 added new model U-net Face Morpher.
removed AVATAR - useless model was just for demo
removed MIAEF128 - use UFM insted
removed LIAEF128YAW - use model option sort by yaw on start for any model
All models now ask some options on start.
Session options (such as target epoch, batch_size, write_preview_history etc) can be overrided by special command arg.
Converter now always ask options and no more support to define options via command line.
fix bug when ConverterMasked always used not predicted mask.
SampleGenerator now always generate samples with replicated border, exclude mask samples.
refactorings
2019-01-02 17:26:12 +04:00
iperov
f3782a012b
Update Trainer.py 2019-01-01 19:50:31 +04:00
iperov
5c0c79e528 fix last PR 2019-01-01 19:42:19 +04:00
iperov
c707fc0658
Update Trainer.py 2019-01-01 18:09:27 +04:00
Plucky
a8694b73f0 DockerFile for Mac users to run DeepfaceLab with CPU Mode (#95)
* fix localization nullpointer exception

* fix devicelib error line:61,remove e

* support create docker from cpu dockerfile

* support preview or not when train(resolve cannot connect to X server)
2019-01-01 18:08:21 +04:00
iperov
6bd5a44264 initial 2018-06-04 17:12:43 +04:00