Commit Graph
100 Commits
Author SHA1 Message Date
iperov 093ca4dba6 fix options in ModelBase.py 2019-04-21 21:08:45 +04:00
iperov 63beb3afd2 SAE: revert back df model from prev commit. LIAE should be restarted. 2019-04-21 08:31:05 +04:00
iperov 2cdf2745a2 SAE: model will not work with this update, you should to restart training,
applied paper https://arxiv.org/abs/1807.01442 https://github.com/aditya-grover/uae
2019-04-20 22:33:28 +04:00
iperov 7c5f762e78 SAE:fix multiscale decoder 2019-04-20 21:30:40 +04:00
iperov 242b850245 fix bgr_shuffle randomizer 2019-04-20 18:05:11 +04:00
iperov dcbdc79e84 fix for plaidML 2019-04-20 17:29:50 +04:00
iperov 8e16f4fa1b fix 2019-04-20 16:10:58 +04:00
iperov 3ffdc50576 nothing interesting 2019-04-20 15:22:00 +04:00
iperovandGitHub 046649e6be update == 04.20.2019 == (#242)
* superb improved fanseg

* _

* _

* added FANseg extractor for src and dst faces to use it in training

* -

* _

* _

* update to 'partial' func

* _

* trained FANSeg_256_full_face.h5,
new experimental models: AVATAR, RecycleGAN

* _

* _

* _

* fix for TCC mode cards(tesla), was conflict with plaidML initialization.

* _

* update manuals

* _
2019-04-20 08:23:08 +04:00
iperov 7be2fd67f5 added chinesse community site 2019-04-11 18:27:12 +04:00
iperov 2276f5eb73 fix zoom keys 2019-04-08 17:35:06 +04:00
iperov 8561c1a931 fix MaskEditor, upd manual ru 2019-04-08 14:13:36 +04:00
iperov 7c930121e9 MaskEditor: added window zoom 2019-04-08 10:30:59 +04:00
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
iperovandGitHub 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
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
iperov cbe2ebca7f SAE: removed random normal initialization 2019-03-23 14:57:30 +04:00
iperov 580e8250e0 fix req for new plaidml 2019-03-21 21:29:35 +04:00
iperov c4f41a7e76 Converter: added new mask modes: FAN-prd, FAN-dst 2019-03-21 21:19:03 +04:00
iperov 9849bcc1e5 fix 2019-03-21 19:04:46 +04:00
iperov 48314f74db removing dlib detector that is outdated and inaccurate 2019-03-21 19:00:42 +04:00
iperov 565af4d1da refactoring 2019-03-21 18:58:38 +04:00
iperov af1760b22d plaidML: upgrading version 2019-03-21 18:57:02 +04:00
iperov 7a0cc56603 extractor: fixes, optimizations,
manual extractor: added 'a' option to switch accuracy mode
2019-03-21 18:56:32 +04:00
iperov 63c9be3d1f fix 2019-03-21 12:15:04 +04:00
iperov d1676dea06 refactorings, optimizations 2019-03-21 11:58:54 +04:00
iperov 97008291b8 fix 2019-03-20 14:43:02 +04:00
iperov 6169e6ba8a fix fanseg 2019-03-20 09:08:42 +04:00
iperov a3df04999c removing trailing spaces 2019-03-19 23:53:27 +04:00
iperov fa4e579b95 fix fanseg 2019-03-19 23:49:41 +04:00
iperov 034ad3cce5 upd fan segmentator 2019-03-19 19:44:14 +04:00
iperov 1f569117c8 accurate landmark extractor now by default 2019-03-19 14:01:15 +04:00
iperov a0c1ce3e7e final sort: increased speed 2019-03-19 13:45:26 +04:00
iperov 33606e327b fixed landmark extraction of face with opened mouth 2019-03-19 00:23:52 +04:00
iperov 7c1ea1a617 removing dflimg image_to_face_mat 2019-03-18 14:42:10 +04:00
iperov 5d819e3e83 upd manual_ru 2019-03-18 11:49:01 +04:00
iperov 00c6a0cdaf accurate landmarks extraction now as an option 2019-03-18 11:24:41 +04:00
iperov bf831331e6 fix 2019-03-18 10:58:03 +04:00
iperov da7cf225db fix 2019-03-18 10:38:48 +04:00
iperov b8efb4cbba enhancing landmarks extractor by using s3fd second pass inside second pass,
it will be x2 slower, but time will be saved due to more images will be marked properly
works on 2GB+
2019-03-18 10:25:24 +04:00
iperov 1d56585f33 fix ability for auto re-extract dst faces, after this update DFL torrent extractor scripts will not work. 2019-03-18 01:12:04 +04:00
iperov b03d62a29b extractor: added ability to re-extract dst faces with transferring all necessary meta data 2019-03-18 00:43:07 +04:00
iperov 00e56299c4 S3FD: wrong small face check fix 2019-03-18 00:30:08 +04:00
iperov 2b40fa182f fix 2019-03-17 23:50:18 +04:00
iperov a71defc69a removing fail solutions 2019-03-17 23:47:14 +04:00
iperov c311f09fa4 _ 2019-03-17 21:14:24 +04:00
iperov 73e91cc0b5 upd LandmarksProcessor.py 2019-03-17 21:14:05 +04:00
iperov 7aebfa3f7f added AdaBound 2019-03-17 19:17:44 +04:00
iperov a7bec17a34 _ 2019-03-17 16:24:36 +04:00
iperov d71a310fd7 SAE: forgot to remove normalizing from tanh 2019-03-17 13:03:49 +04:00
iperov 4ff88865ce refactoring 2019-03-16 21:29:56 +04:00
iperov f009edae79 FANSegmentator for developing 2019-03-16 20:58:20 +04:00
iperov 1c8845d558 model fansegmentator for developing purposes 2019-03-16 20:57:51 +04:00
iperov f3b343c0e5 small fixed and refactorings 2019-03-16 20:55:51 +04:00