Commit Graph
100 Commits
Author SHA1 Message Date
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
iperov d6a45763a2 SAE: added option "Use CA weights":
Initialize network with 'Convolution Aware' weights. This may help to achieve a higher accuracy model, but consumes time at first run.
2019-03-16 12:54:36 +04:00
iperov 71ff0ce1a7 nothing interesting 2019-03-16 09:22:00 +04:00
iperov 7c66503382 sorter: fix final sort 2019-03-16 09:20:55 +04:00
iperov 5076feb90f SAE: revert back to sigmoid. Trainings must be restarted. 2019-03-16 09:01:11 +04:00
iperov ea5fe28d08 fix final sorter 2019-03-15 23:41:34 +04:00
iperov a8c987fa0f fix performance issue 2019-03-15 11:50:13 +04:00
iperov 4ffae1a954 fix performance issue 2019-03-15 11:48:38 +04:00
iperov 3549272244 fix for final 2019-03-15 11:03:37 +04:00
iperov 1beb2f07f0 fix get_image_hull_mask 2019-03-14 15:43:10 +04:00
iperov 9823421a44 added transparent mask to draw_landmarks 2019-03-14 12:16:21 +04:00
iperov 5bd3f875fa fix 2019-03-14 10:50:53 +04:00
iperov 6e59f1a150 Sorter: new better blur estimator 2019-03-14 10:38:56 +04:00
iperov 296e8a111a fix for Linux 2019-03-14 05:08:15 +04:00
iperov 62af1b6f16 S3FD : filtering faces < 40pix. 2019-03-14 01:23:39 +04:00
iperov 8da47fec13 fix ModelBase, nnlib 2019-03-13 20:53:59 +04:00
iperov da0fc2d2e1 update manual_ru 2019-03-13 20:03:46 +04:00
iperov 3375088669 SAE: optimizer_mode option now cannot be overridden by restart 2019-03-13 20:00:16 +04:00
iperov a9026ccb67 fix ModelBase, nnlib 2019-03-13 19:50:16 +04:00
iperov 90a7d4b1e7 fix force_plaidML 2019-03-13 18:30:40 +04:00
iperov 58763756f5 SAE: removed simple_optimizer . Added optimizer mode for tensorflow only (NVIDIA cards), allows to train x2-x3 bigger networks with normal Adam optimizer, consuming VRAM and CPU power. 2019-03-13 11:54:17 +04:00
iperov 3c6775bae7 fix ModelBase.py 2019-03-12 23:22:58 +04:00
iperov aaf7ecd595 update manual_ru 2019-03-12 21:27:37 +04:00
iperov 97b6fabaab change 'epoch' to 'iter',
added timestamp prefix to training string
2019-03-12 19:23:52 +04:00
iperov 69174a48e0 fix DFLOptimizer 2019-03-12 16:56:59 +04:00
iperov 46ff33bf89 SAE: dssim kernel size now depends on resolution 2019-03-12 09:49:40 +04:00
iperov fd3b9add2f SAE: added option "simple optimizer" allows to train bigger networks on same VRAM
nnlib: added DFLOptimizer is my own optimizer
2019-03-12 09:32:35 +04:00
iperov 3bad8dd8ec fix 2019-03-11 21:53:28 +04:00
iperov ee8dbcbc35 revert back Adam 2019-03-11 21:52:36 +04:00
iperov e4637336ef added ability to save optimizers states which work with K.function,
added custom Adam that can save 'iterations' param
2019-03-11 18:23:01 +04:00
iperov e50dc0d748 2DFAN-4: fix bug 2019-03-11 10:26:03 +04:00
iperov 52f149a570 update manual_ru 2019-03-11 09:34:47 +04:00
iperov 893354b870 fix S3FD for plaidML, fix S3FD warnings on Lambda layers. 2019-03-11 09:32:00 +04:00
iperov 338a8fccc6 added missing file 2019-03-10 23:41:02 +04:00
iperov fbf39d2727 added new extractor: S3FD,
all extractors now produce less false-positive faces
2019-03-10 23:18:10 +04:00
iperov 9440224556 change TF console log level to 'errors only' 2019-03-09 20:23:33 +04:00
iperov 95e70d510f upd doc_prebuilt_windows_app.md 2019-03-09 10:09:27 +04:00
iperov f2de3b5ab7 upd manual_ru 2019-03-09 10:02:14 +04:00
iperov 9cc1a20255 update manual_ru 2019-03-08 20:15:00 +04:00
iperov 168d48b119 extractor: increased speed of 1st pass by DLIB for >= 4GB vram cards,
added 'h' key for hide the help in manual extractor
2019-03-08 17:01:11 +04:00
iperov bced4d3259 - 2019-03-08 16:39:27 +04:00
iperov 0cc251a9ab 2nd pass of extractor : speed significantly increased 2019-03-08 16:05:24 +04:00
iperov 8f7b268376 upd manual_ru 2019-03-07 20:51:37 +04:00
iperov e625cfef0b upd doc/doc_prebuilt_windows_app.md 2019-03-07 20:39:53 +04:00
iperov 5587c93e01 fix ConverterMasked.py,
changing requirements
changing device.py ENV vars
2019-03-07 19:48:12 +04:00
iperov c3eea0cf98 upd device.py 2019-03-04 11:17:08 +04:00
iperov bdb45357ee rename requirements-AVX-cuda10.0-cudnn7.4.1.txt 2019-03-04 10:58:52 +04:00
iperov 6094e5ead8 added Util.recover_original_aligned_filename 2019-03-04 10:57:44 +04:00