Commit graph

39 commits

Author SHA1 Message Date
Colombo
dbf28eeebf fix for linux 2020-02-29 14:36:10 +04:00
Colombo
5eca4958b7 fix for linux 2020-02-29 14:34:08 +04:00
Colombo
fcbfbdd560 _ 2020-02-29 14:31:21 +04:00
Colombo
7e276a4ec7 _ 2020-02-29 14:12:53 +04:00
Colombo
7180076ecf fix 2020-02-29 13:31:36 +04:00
Colombo
90ae68ae61 fix input_in_time for linux 2020-02-29 11:05:04 +04:00
Colombo
59be114485 upd leras 2020-02-27 16:20:14 +04:00
Colombo
9860a38907 upd SampleGenerator 2020-02-27 09:58:46 +04:00
Colombo
1898bd6881 _ 2020-02-26 23:50:42 +04:00
Colombo
f07bbd5fb0 update leras 2020-02-26 13:32:32 +04:00
Colombo
0324592b67 update leras/layers.py 2020-02-24 22:23:08 +04:00
Colombo
f1d115b63b added experimental face type 'whole_face'
Basic usage instruction: https://i.imgur.com/w7LkId2.jpg

	'whole_face' requires skill in Adobe After Effects.

	For using whole_face you have to extract whole_face's by using
	4) data_src extract whole_face
	and
	5) data_dst extract whole_face
	Images will be extracted in 512 resolution, so they can be used for regular full_face's and half_face's.

	'whole_face' covers whole area of face include forehead in training square,
	but training mask is still 'full_face'
	therefore it requires manual final masking and composing in Adobe After Effects.

added option 'masked_training'
	This option is available only for 'whole_face' type.
	Default is ON.
	Masked training clips training area to full_face mask,
	thus network will train the faces properly.
	When the face is trained enough, disable this option to train all area of the frame.
	Merge with 'raw-rgb' mode, then use Adobe After Effects to manually mask, tune color, and compose whole face include forehead.
2020-02-21 16:21:04 +04:00
Colombo
7a790489f2 fix SubprocessorBase.py 2020-02-19 22:09:45 +04:00
Colombo
7a9cd6379f bug fixes 2020-02-19 15:27:22 +04:00
Colombo
6aa585fa33 fix leras 2020-02-19 07:00:29 +04:00
Colombo
9598ba0141 SAEHD:
added option Eyes priority (y/n)

	fix eye problems during training  ( especially on HD architectures )
	by forcing the neural network to train eyes with higher priority
	before/after https://i.imgur.com/YQHOuSR.jpg

	It does not guarantee the right eye direction.
2020-02-18 14:30:07 +04:00
Colombo
01376fd17c decreased time of training initialization 2020-02-18 10:01:33 +04:00
Colombo
e0a55ff1c3 update leras 2020-02-17 18:26:19 +04:00
Colombo
b24b1e2950 Merger: fixes and better quality 2020-02-01 04:42:43 +04:00
Colombo
3f813d5611 refactoring 2020-01-31 14:31:26 +04:00
Colombo
80036f950f leras ModelBase.summary() 2020-01-28 21:38:15 +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
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
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
f1a3150ce9 1 2020-01-22 22:14:31 +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
Colombo
38b85108b3 DFL-2.0 initial branch commit 2020-01-21 18:43:39 +04:00