Commit graph

1311 commits

Author SHA1 Message Date
Colombo
27f219fec7 XSegEditor: fixed incorrect count of labeled images. 2020-06-21 00:56:25 +04:00
Colombo
2b8e8f0554 . 2020-06-20 22:05:30 +04:00
Colombo
53222839c6 SAEHD: for -d archis resolution should be divided by 32 2020-06-19 14:21:41 +04:00
Colombo
05606b5e15 Subprocessor: now should write an error if failed to start 2020-06-19 14:20:22 +04:00
Colombo
0c2e1c3944 SAEHD:
Maximum resolution is increased to 640.

‘hd’ archi is removed. ‘hd’ was experimental archi created to remove subpixel shake, but ‘lr_dropout’ and ‘disable random warping’ do that better.

‘uhd’ is renamed to ‘-u’
dfuhd and liaeuhd will be automatically renamed to df-u and liae-u in existing models.

Added new experimental archi (key -d) which doubles the resolution using the same computation cost.
It is mean same configs will be x2 faster, or for example you can set 448 resolution and it will train as 224.
Strongly recommended not to train from scratch and use pretrained models.

New archi naming:
'df' keeps more identity-preserved face.
'liae' can fix overly different face shapes.
'-u' increased likeness of the face.
'-d' (experimental) doubling the resolution using the same computation cost
Examples: df, liae, df-d, df-ud, liae-ud, ...

Improved GAN training (GAN_power option).  It was used for dst model, but actually we don’t need it for dst.
Instead, a second src GAN model with x2 smaller patch size was added, so the overall quality for hi-res models should be higher.

Added option ‘Uniform yaw distribution of samples (y/n)’:
	Helps to fix blurry side faces due to small amount of them in the faceset.

Quick96:
	Now based on df-ud archi and 20% faster.

XSeg trainer:
	Improved sample generator.
Now it randomly adds the background from other samples.
Result is reduced chance of random mask noise on the area outside the face.
Now you can specify ‘batch_size’ in range 2-16.

Reduced size of samples with applied XSeg mask. Thus size of packed samples with applied xseg mask is also reduced.
2020-06-19 09:45:55 +04:00
Colombo
9fd3a9ff8d add meme 2020-06-18 22:15:25 +04:00
Colombo
38573473da added youtube channel 2020-06-18 14:19:58 +04:00
Colombo
9994f1512e added third head swap example 2020-06-18 13:56:42 +04:00
Colombo
11327cb0c5 upd face examples 2020-06-17 10:51:11 +04:00
Colombo
70182f8c65 upd face examples 2020-06-17 10:49:41 +04:00
Colombo
8c467bad90 fuck linux 2020-06-15 11:26:22 +04:00
Colombo
2e650a7122 . 2020-06-12 10:45:57 +04:00
Colombo
7f11713730 fix 2020-06-11 21:29:39 +04:00
Colombo
d99afebbdc new telegram group link 2020-06-11 17:39:36 +04:00
Colombo
82f405ed49 Trainer: fixed "Choose image for the preview history". Now you can switch between subpreviews using 'space' key.
Fixed "Write preview history". Now it writes all subpreviews in separated folders
https://i.imgur.com/IszifCJ.jpg
also the last preview saved as _last.jpg before the first file
https://i.imgur.com/Ls1AOK4.jpg
thus you can easily check the changes with the first file in photo viewer
2020-06-10 10:20:13 +04:00
Colombo
5c315cab68 XSegEditor: added text label of total labeled images
Changed frame line design
Changed loading frame design
2020-06-09 22:04:15 +04:00
Colombo
dc43f5a891 add second headswap example 2020-06-09 16:01:47 +04:00
Colombo
c2b5f80e3f upd readme download link 2020-06-08 15:03:40 +04:00
Colombo
1d3d417aee upd readme 2020-06-08 14:53:19 +04:00
Colombo
af98407f06 SAEHD: lr_dropout now can be ‘n’, ‘y’, ‘cpu’. ‘n’ and ’y’ are the same as before.
‘cpu’ mean enabled on CPU. This allows not to use extra VRAM, sacrificing 20% time of iteration.

SAEHD: resolution >= 256 now has second dssim loss function
2020-06-08 14:18:33 +04:00
Colombo
4acaf08ebf fix 2020-06-08 14:07:47 +04:00
Colombo
cfd7803e0d leras : DepthwiseConv2D 2020-06-07 20:59:12 +04:00
Colombo
85ede1a7e7 . 2020-06-07 12:26:23 +04:00
Colombo
ddf7363eda remove unused code 2020-06-06 20:02:21 +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
61f93063b2 fix 2020-06-04 16:44:29 +04:00
Colombo
addc96fe3e fix 2020-06-04 16:42:36 +04:00
Ilya Sergeev
4ed320a86b
fix nan in multiply (#769)
Co-authored-by: Ilya Sergeev
2020-06-04 16:39:46 +04:00
Colombo
65bc9273e6 upd readme 2020-06-04 15:38:20 +04:00
Colombo
c94c9106be upd readme 2020-06-04 14:44:47 +04:00
Colombo
903657ff91 upd readme 2020-06-04 14:39:50 +04:00
Colombo
f91a604c6d Manual extractor: now you can specify face rectangle manually using ‘R Mouse button’.
It is useful for small, blurry, undetectable faces, animal faces.
https://i.imgur.com/8kmVgg8.jpg
Warning: such frames must be used only with XSeg workflow !
Landmarks cannot be placed on the face precisely, and they are actually used for positioning the red frame.
Try to keep the red frame the same as the adjacent frames.
2020-06-02 12:31:24 +04:00
Colombo
e77865ce18 XSegEditor: fix 2020-06-01 13:44:29 +04:00
Colombo
6f25ebbc9e XSegEditor: added button "view XSeg mask overlay face" 2020-05-31 18:00:39 +04:00
Colombo
c6652ac006 add arxiv paper 2020-05-24 15:10:40 +04:00
Colombo
130d72cb8b upd 2020-05-22 22:44:00 +04:00
Colombo
72e2f771a2 upd readme 2020-05-20 09:58:55 +04:00
Colombo
0c18b1f011 changed uhd arhi: you have to retrain uhd arhis from scratch. 2020-05-06 19:21:23 +04:00
Colombo
dcf380345b fix 2020-05-02 14:04:32 +04:00
Colombo
f935bf0465 . 2020-04-28 19:04:44 +04:00
Colombo
7b49220ee9 fix 2020-04-20 18:40:12 +04:00
Colombo
4fe45adc7c . 2020-04-20 10:24:51 +04:00
Colombo
3a9e851339 fix bug 2020-04-18 18:29:55 +04:00
Colombo
a3c91271d0 upd README.md : google drive is replaced by mega 2020-04-16 00:08:42 +04:00
Colombo
baff59e421 . 2020-04-15 22:58:59 +04:00
Colombo
ea607edfc9 XSegEditor: added view lock at the center by holding shift. 2020-04-15 22:12:54 +04:00
Colombo
6bbc607312 color transfer sot : optimization 2020-04-15 18:13:03 +04:00
Colombo
f8580928ed SampleLoader: fix error 2020-04-15 16:55:51 +04:00
Colombo
c42b1b2124 . 2020-04-15 10:44:12 +04:00