Commit graph

1311 commits

Author SHA1 Message Date
Colombo
e0a55ff1c3 update leras 2020-02-17 18:26:19 +04:00
Colombo
e1635ff760 added CentOS Linux fork 2020-02-15 16:39:56 +04:00
Colombo
60cc917350 add eye masking code 2020-02-03 06:38:58 +04:00
Colombo
4850ef4965 autobackup hour 2020-02-03 06:37:54 +04:00
Colombo
5620763ccf "Enable autobackup" option is replaced by
"Autobackup every N hour" 0..24 (default 0 disabled), Autobackup model files with preview every N hour
2020-02-02 20:53:18 +04:00
Colombo
ee27bb0aac Merger: fix bug 2020-02-02 14:45:48 +04:00
Colombo
85792d232a Merger:
'show alpha mask' now on 'V' button

'super resolution mode' is replaced by
'super resolution power' (0..100) which can be modified via 'T' 'G' buttons
2020-02-02 10:13:23 +04:00
Colombo
054a7e0e6e Merger: now uses all available CPU cores ( before max 6 )
so the more processors, the faster the process will be.
2020-02-01 18:58:10 +04:00
Colombo
9c07983c68 Merger:
default erode/blur values are 0.

new multiple faces detection log: https://i.imgur.com/0XObjsB.jpg
2020-02-01 18:50:43 +04:00
Colombo
1a0d7846a0 fix 2020-02-01 04:46:34 +04:00
Colombo
b24b1e2950 Merger: fixes and better quality 2020-02-01 04:42:43 +04:00
Colombo
05f32d678d SAEHD: default archi is now 'df' 2020-01-31 22:42:25 +04:00
Colombo
430c2f974e upd README.md 2020-01-31 22:32:21 +04:00
Colombo
123c015fdc Merger:
increased speed

improved quality
2020-01-31 22:22:32 +04:00
Colombo
3f813d5611 refactoring 2020-01-31 14:31:26 +04:00
Colombo
4ed16c67f5 upd readme 2020-01-31 14:24:03 +04:00
Colombo
126c2bce5c _ 2020-01-31 09:41:54 +04:00
Colombo
3e63ba2d17 fix mean loss history after save 2020-01-30 07:49:45 +04:00
Colombo
9fd49ee3f0 removed use_float16 option
fix multigpu training
2020-01-30 07:35:33 +04:00
Colombo
5fe5fa131c SampleProcessor.py : refactoring and gen mask struct 2020-01-29 18:08:54 +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
a6d72b620d fix asking options 2020-01-29 02:40:57 +04:00
Colombo
80036f950f leras ModelBase.summary() 2020-01-28 21:38:15 +04:00
Colombo
1fd4a25882 use_float16 update description 2020-01-28 15:22:03 +04:00
Colombo
01800010d4 options order fix 2020-01-28 15:17:11 +04:00
Colombo
9c6ca24642 fix update preview samples after disable pretrain 2020-01-28 13:32:01 +04:00
Colombo
7386a9d6fd optimized face sample generator, CPU load is significantly reduced
SAEHD:

added new option
GAN power 0.0 .. 10.0
	Train the network in Generative Adversarial manner.
	Forces the neural network to learn small details of the face.
	You can enable/disable this option at any time,
	but better to enable it when the network is trained enough.
	Typical value is 1.0
	GAN power with pretrain mode will not work.

Example of enabling GAN on 81k iters +5k iters
https://i.imgur.com/OdXHLhU.jpg
https://i.imgur.com/CYAJmJx.jpg

dfhd: default Decoder dimensions are now 48
the preview for 256 res is now correctly displayed

fixed model naming/renaming/removing

Improvements for those involved in post-processing in AfterEffects:

Codec is reverted back to x264 in order to properly use in AfterEffects and video players.

Merger now always outputs the mask to workspace\data_dst\merged_mask

removed raw modes except raw-rgb
raw-rgb mode now outputs selected face mask_mode (before square mask)

'export alpha mask' button is replaced by 'show alpha mask'.
You can view the alpha mask without recompute the frames.

8) 'merged *.bat' now also output 'result_mask.' video file.
8) 'merged lossless' now uses x264 lossless codec (before PNG codec)
result_mask video file is always lossless.

Thus you can use result_mask video file as mask layer in the AfterEffects.
2020-01-28 12:24:45 +04:00
Colombo
80f285067a upd readme 2020-01-27 20:30:07 +04:00
TooMuchFun
9b6bb4cd34 fix io not imported (#580) 2020-01-27 06:56:03 +04:00
Colombo
17337ea8f8 fixed model renaming/deleting 2020-01-26 14:49:37 +04:00
Colombo
c485e1718a fixes 2020-01-26 12:56:21 +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
a3dfcb91b9 1 2020-01-25 18:18:06 +04:00
Colombo
f7f4e44a98 1 2020-01-24 11:51:54 +04:00
Colombo
be5df1e269 fix 2020-01-23 22:46:45 +04:00
Colombo
1f0c91f053 add SampleGeneratorFaceTemporal.py 2020-01-23 19:09:23 +04:00
Colombo
afcfa9c69e fix io.input_skip_pending for colab 2020-01-23 10:20:29 +04:00
Colombo
591bba2c1e fix clipgrad option 2020-01-23 09:30:43 +04:00
Colombo
967fbdd814 BREAKING CHANGES !!!
Getting rid of the weakest link - AMD cards support.
All neural network codebase transferred to pure low-level TensorFlow backend, therefore
removed AMD/Intel cards support, now DFL works only on NVIDIA cards or CPU.

old DFL marked as 1.0 still available for download, but it will no longer be supported.

global code refactoring, fixes and optimizations

Extractor:

now you can choose on which GPUs (or CPU) to process

improved stability for < 4GB GPUs

increased speed of multi gpu initializing

now works in one pass (except manual mode)
so you won't lose the processed data if something goes wrong before the old 3rd pass

Faceset enhancer:

now you can choose on which GPUs (or CPU) to process

Trainer:

now you can choose on which GPUs (or CPU) to train the model.
Multi-gpu training is now supported.
Select identical cards, otherwise fast GPU will wait slow GPU every iteration.

now remembers the previous option input as default with the current workspace/model/ folder.

the number of sample generators now matches the available number of processors

saved models now have names instead of GPU indexes.
Therefore you can switch GPUs for every saved model.
Trainer offers to choose latest saved model by default.
You can rename or delete any model using the dialog.

models now save the optimizer weights in the model folder to continue training properly

removed all models except SAEHD, Quick96

trained model files from DFL 1.0 cannot be reused

AVATAR model is also removed.
How to create AVATAR like in this video? https://www.youtube.com/watch?v=4GdWD0yxvqw
1) capture yourself with your own speech repeating same head direction as celeb in target video
2) train regular deepfake model with celeb faces from target video as src, and your face as dst
3) merge celeb face onto your face with raw-rgb mode
4) compose masked mouth with target video in AfterEffects

SAEHD:

now has 3 options: Encoder dimensions, Decoder dimensions, Decoder mask dimensions

now has 4 arhis: dfhd (default), liaehd, df, liae
df and liae are from SAE model, but use features from SAEHD model (such as combined loss and disable random warp)

dfhd/liaehd - changed encoder/decoder architectures

decoder model is combined with mask decoder model
mask training is combined with face training,
result is reduced time per iteration and decreased vram usage by optimizer

"Initialize CA weights" now works faster and integrated to "Initialize models" progress bar

removed optimizer_mode option

added option 'Place models and optimizer on GPU?'
  When you train on one GPU, by default model and optimizer weights are placed on GPU to accelerate the process.
  You can place they on CPU to free up extra VRAM, thus you can set larger model parameters.
  This option is unavailable in MultiGPU mode.

pretraining now does not use rgb channel shuffling
pretraining now can be continued
when pre-training is disabled:
1) iters and loss history are reset to 1
2) in df/dfhd archis, only the inter part of the encoder is reset (before encoder+inter)
   thus the fake will train faster with a pretrained df model

Merger ( renamed from Converter ):

now you can choose on which GPUs (or CPU) to process

new hot key combinations to navigate and override frame's configs

super resolution upscaler "RankSRGAN" is replaced by "FaceEnhancer"

FAN-x mask mode now works on GPU while merging (before on CPU),
therefore all models (Main face model + FAN-x + FaceEnhancer)
now work on GPU while merging, and work properly even on 2GB GPU.

Quick96:

now automatically uses pretrained model

Sorter:

removed all sort by *.bat files except one sort.bat
now you have to choose sort method in the dialog

Other:

all console dialogs are now more convenient

new default example video files data_src/data_dst for newbies ( Robert Downey Jr. on Elon Musk )

XnViewMP is updated to 0.94.1 version

ffmpeg is updated to 4.2.1 version

ffmpeg: video codec is changed to x265

_internal/vscode.bat starts VSCode IDE where you can view and edit DeepFaceLab source code.

removed russian/english manual. Read community manuals and tutorials here
https://mrdeepfakes.com/forums/forum-guides-and-tutorials

new github page design
2020-01-22 22:23:55 +04:00
Colombo
f1a3150ce9 1 2020-01-22 22:14:31 +04:00
Colombo
869c5e8dd0 1 2020-01-22 18:58:02 +04:00
Colombo
af3306a85a 1 2020-01-22 18:56:00 +04:00
Colombo
0ce96875a1 1 2020-01-22 18:49:32 +04:00
Colombo
2d61b715de 1 2020-01-22 18:27:19 +04:00
Colombo
afafd89982 1 2020-01-22 17:23:08 +04:00
Colombo
3924ef2bf5 1 2020-01-22 17:21:33 +04:00
Colombo
60804ca3ba 1 2020-01-22 13:41:05 +04:00
Colombo
0c93b89e87 1 2020-01-22 10:46:35 +04:00
Colombo
beed145d29 1 2020-01-22 10:29:17 +04:00
Colombo
9797a70fd3 1 2020-01-21 21:05:29 +04:00