Commit graph

263 commits

Author SHA1 Message Date
Colombo
9365e42f25 trainer now creates directories instead of throwing error when they do not exist 2020-03-04 17:54:52 +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
72797ce406 linux support ffmpeg aac "strict": "experimental" 2020-02-29 08:55:17 +04:00
Colombo
810dc5ada5 Extractor: image size for all faces now 512 2020-02-27 16:20:37 +04:00
Colombo
a5783df546 fix autobackup_hour 2020-02-27 11:13:45 +04:00
Colombo
b3b97b9802 fix RuntimeWarning during the extraction process 2020-02-25 09:59:48 +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
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
20f842a34a Merger no more terminate process after 60 secs 2020-02-19 22:11:28 +04:00
Colombo
4f928074b9 removing smooth_rect option 2020-02-18 10:28:01 +04:00
Colombo
814da70577 Merger:
added smooth_rect option
	default is ON.
	Decreases jitter of predicting rect by using temporal interpolation.
	You can disable this option if you have problems with dynamic scenes.
2020-02-17 18:27:09 +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
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
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
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
f1a3150ce9 1 2020-01-22 22:14:31 +04:00
Colombo
0ce96875a1 1 2020-01-22 18:49:32 +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
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
f95682f616 fix 2020-01-02 21:19:19 +04:00
Colombo
4e67278c66 fix 0 faces detected in manual extractor 2019-12-28 16:45:40 +04:00
Colombo
7804e0bd7c add try except 2019-12-27 13:36:01 +04:00
Colombo
df8cf35c75 FacesetEnhancer now asks to merge aligned_enhanced/ to aligned/ 2019-12-27 11:09:15 +04:00
Colombo
d46fb5cfd3 fixed mask editor
added FacesetEnhancer
4.2.other) data_src util faceset enhance best GPU.bat
4.2.other) data_src util faceset enhance multi GPU.bat

FacesetEnhancer greatly increases details in your source face set,
same as Gigapixel enhancer, but in fully automatic mode.
In OpenCL build it works on CPU only.

Please consider a donation.
2019-12-26 21:27:10 +04:00
Colombo
3be223a265 fix MaskEditor 2019-12-25 11:13:52 +04:00
Colombo
7174dc835a fixes 2019-12-22 21:44:28 +04:00
Colombo
754d6c385c refactorings 2019-12-22 19:00:59 +04:00
Colombo
e0e8970ab9 fix PackedFaceset 2019-12-22 15:58:46 +04:00
Colombo
dd45b7dacc cleaning 2019-12-22 15:02:53 +04:00
Colombo
8153e90ba3 refactoring 2019-12-22 14:05:41 +04:00
Colombo
9312a3c334 Extractor: 2nd pass now faster on frames where faces are not found 2019-12-22 10:46:44 +04:00
Colombo
8866dce22e added
4.2.other) data_src util faceset metadata save.bat
	saves metadata of data_src\aligned\ faces into data_src\aligned\meta.dat

4.2.other) data_src util faceset metadata restore.bat
	restore metadata from 'meta.dat' to images
	if image size different from original, then it will be automatically resized

You can greatly enhance face details of src faceset by using Topaz Gigapixel software.
example https://i.imgur.com/Gwee99L.jpg
Example of workflow:
1) run 'data_src util faceset metadata save.bat'
2) launch Topaz Gigapixel
3) open 'data_src\aligned\' and select all images
4) set output folder to 'data_src\aligned_topaz' (create folder in save dialog)
5) set settings as on screenshot https://i.imgur.com/kAVWMQG.jpg
	you can choose 2x, 4x, or 6x upscale rate
6) start process images and wait full process
7) rename folders:
	data_src\aligned        ->  data_src\aligned_original
	data_src\aligned_topaz  ->  data_src\aligned
8) copy 'data_src\aligned_original\meta.dat' to 'data_src\aligned\'
9) run 'data_src util faceset metadata restore.bat'
	images will be downscaled back to original size (256x256) preserving details
	metadata will be restored
10) now your new enhanced faceset is ready to use !
2019-12-20 15:03:17 +04:00
Colombo
28b948888c fix extractor for some systems 2019-12-17 09:52:48 +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
1bfd65abe5 extractor for AMD gpu: increase no response time 2019-11-24 19:37:21 +04:00
Colombo
6b14741160 1 2019-11-21 11:40:55 +04:00
Colombo
668513a507 1 2019-11-16 16:54:10 +04:00
Colombo
006233656b facerelighther: added intensity param 2019-11-13 00:19:26 +04:00
Colombo
8c3fa3baa5 fix 2019-11-12 18:02:23 +04:00