Commit graph

627 commits

Author SHA1 Message Date
iperov
58d7e990f4 refactorings, added motion blur to SampleProcessor for FANSegmentator trainer 2019-04-07 23:08:00 +04:00
iperov
a88ee7d093 fix 2019-04-06 17:48:33 +04:00
iperov
43d68e1307 MaskEditor: added prev/next frames preview 2019-04-06 16:22:48 +04:00
iperov
9f68b25987 Extractor: added warning messages if aligned folder contains files that will be deleted. 2019-04-06 10:01:59 +04:00
iperov
cfae9f9fd8 trainer: fix for ctrl-c in --no-preview mode 2019-04-05 21:50:46 +04:00
iperov
fe40a46933 limiting converter subprocesses to max 6 2019-04-05 21:43:24 +04:00
iperov
1bd864cc15 fix imagelib\text.py 2019-04-05 21:42:10 +04:00
iperov
41601f50b8 upd pynvml.py 2019-04-05 21:33:53 +04:00
iperov
406ef91851 fix mask editor 2019-04-05 14:42:11 +04:00
iperov
d3ac88b799 fix MaskEditorTool 2019-04-05 09:32:04 +04:00
iperov
de8d75b4f7 fix mask editor,
upd manuals
2019-04-04 22:19:56 +04:00
iperov
0fa9d269ec fixes 2019-04-04 16:34:45 +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
01e98cde8e upd pixel_loss help message 2019-03-31 19:21:04 +04:00
iperov
2d4773b18f _ 2019-03-31 17:24:07 +04:00
iperov
5f811a1b75 remove asking color transfer in raw mode 2019-03-31 16:48:36 +04:00
iperov
178fd070f7 Converter: fix blur region of seamless. 2019-03-31 08:52:06 +04:00
iperov
c1117852d8 +imagelib.normalize_channels 2019-03-30 21:52:36 +04:00
iperov
a885838363 make seamless great again!
fixed seamless face jitter
removed options Suppress seamless jitter, seamless erode mask modifier.
seamlessed face now properly uses blur modifier
added option 'FAN-prd&dst' - using multiplied FAN prd and dst mask,
2019-03-30 11:58:40 +04:00
iperov
13f3251ee3 converter: fixed suppress seamless jitter 2019-03-30 10:02:04 +04:00
iperov
7b5fb497f0 upd manual 2019-03-29 18:09:33 +04:00
iperov
459d72113f fix conversion speed 2019-03-29 00:18:22 +04:00
iperov
85c01e3b4a Converter: added Apply super resolution? (y/n skip:n) : , Enhance details by applying DCSCN network.
refactorings
2019-03-28 21:50:27 +04:00
iperov
4683c362ac changing SubpixelUpscaler to variable H,W dims,
tensorflow backend : using depth_to_space in SubpixelUpscaler, so training speed increased by 4%
2019-03-28 17:55:42 +04:00
iperov
689aefeb2f Converter: fix for input images with ch > 3 2019-03-28 11:12:19 +04:00
iperov
3cf3bb786e rename samples to samplelib 2019-03-27 10:44:13 +04:00
iperov
773e8d80e0 SampleGeneratorFace added ability to specify random seed (currently unused) 2019-03-27 10:39:21 +04:00
iperov
24dd4ef000 get rid of random module 2019-03-27 10:28:07 +04:00
iperov
f63f78e1ab fix 2019-03-26 19:15:20 +04:00
iperov
808f74e9c8 fix 2019-03-26 19:14:30 +04:00
iperov
77d9f3c856 fix 2019-03-26 19:09:00 +04:00
iperov
fedeee4395 Colab: added ability to run programs after N secs in trainer 2019-03-26 19:06:19 +04:00
iperov
009af42617 fix converter issue #205 2019-03-26 18:23:39 +04:00
iperov
070e938568 colab: final fix for preview filepath 2019-03-26 18:11:33 +04:00
iperov
e8a5a4bf10 _preview.jpg for colab 2019-03-26 17:58:29 +04:00
iperov
6241c8af41 preview file for colab 2019-03-26 17:14:59 +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
79950c6c0e upd readme 2019-03-26 09:16:02 +04:00
iperov
ec0c4d0128 upd readme 2019-03-26 09:15:22 +04:00
lbfs
dbc8faf899 Small typo in argument (#203) 2019-03-25 21:25:47 +04:00
iperov
131b2b5c79 after save loss string now shows averaged value since last save 2019-03-25 18:14:04 +04:00
iperov
bb02cc1d97 _ 2019-03-25 13:28:08 +04:00
iperov
aa58d9e563 SAE: removed lightweight encoder 2019-03-25 10:05:30 +04:00
iperov
addd87efa8 upd README.md 2019-03-24 19:44:09 +04:00
iperov
54239f8edf sae: optimizer mode now can be overrided 2019-03-24 16:25:30 +04:00
iperov
37505d88e3 old SAE model will not work with this update.
Fixed bug when SAE can be collapsed during a time.

SAE: removed CA weights and encoder/decoder dims.

added new options:

Encoder dims per channel (21-85 ?:help skip:%d)
More encoder dims help to recognize more facial features, but require more VRAM. You can fine-tune model size to fit your GPU.

Decoder dims per channel (11-85 ?:help skip:%d)
More decoder dims help to get better details, but require more VRAM. You can fine-tune model size to fit your GPU.

Add residual blocks to decoder? (y/n, ?:help skip:n) :
These blocks help to get better details, but require more computing time.

Remove gray border? (y/n, ?:help skip:n) :
Removes gray border of predicted face, but requires more computing resources.
2019-03-24 15:35:02 +04:00
iperov
4f4447d719 nnlib: implemented ReflectionPadding2D for plaidML,
wrapping Conv2D with new padding param
2019-03-23 22:20:05 +04:00
iperov
673d4915d9 VideoEd: added comment 2019-03-23 18:28:50 +04:00
iperov
668a3e23a1 frame extraction: fixed pixel format for PNG. Added PNG/JPG option. 2019-03-23 18:27:38 +04:00