Commit Graph

953 Commits

Author SHA1 Message Date
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
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