Commit Graph
100 Commits
Author SHA1 Message Date
iperov a299324166 _ 2021-08-12 14:28:36 +04:00
iperov c8e6f23a31 fix SampleProcessor 2021-08-11 10:06:18 +04:00
iperov 4581d48ff0 _ 2021-08-10 20:54:54 +04:00
iperov f99b8a0842 update windows magnet link 2021-07-30 21:38:27 +04:00
iperov 931ec36b41 update windows magnet link 2021-07-30 21:37:37 +04:00
iperov 55b947eab5 XSeg: added pretrain option. 2021-07-30 17:24:21 +04:00
iperov 83b1412da7 Export AMP/SAEHD: added "Export quantized" option 2021-07-30 17:13:46 +04:00
iperov f5cc54177f add related work 2021-07-30 13:45:25 +04:00
iperov d26d579836 AMP change help of ct mode. Default inter dims now 1024 2021-07-29 23:14:30 +04:00
iperov 9e1bc5a153 AMP: return lr_dropout option 2021-07-28 11:00:19 +04:00
iperov 4be135af60 AMP: last high loss samples behaviour - same as SAEHD 2021-07-28 08:58:32 +04:00
iperov bfa88c5fd9 upd requirements-colab.txt 2021-07-19 23:33:56 +04:00
iperov 5c7e2f310a upd readme 2021-07-19 19:58:03 +04:00
iperov c5044580c7 upd readme 2021-07-19 19:55:59 +04:00
iperov 850b1c5cc9 upd readme: added related works section 2021-07-19 19:54:24 +04:00
iperov 493f23c618 update windows magnet link 2021-07-17 23:10:11 +04:00
iperov 9ad9728b40 fix 2021-07-17 22:04:34 +04:00
iperov 336e0ca944 fix 2021-07-17 22:04:20 +04:00
iperov 8b71b83c94 fix 2021-07-17 22:03:59 +04:00
iperov 0748b8d043 AMP: removed eyes mouth prio option(default enabled), removed masked training option(default enabled). 2021-07-17 21:58:57 +04:00
iperov 959a3530f8 AMP: default morph factor is now 0.5. Preview shows with 1.0 2021-07-17 10:27:07 +04:00
iperov da8f33ee85 removing fp16 for amp 2021-07-15 13:23:33 +04:00
iperov c9e0ba1779 removing fp16 for sae 2021-07-15 13:22:32 +04:00
iperov 7b8991a3cc fix help 2021-07-15 00:50:34 +04:00
iperov f044c99ddc remove ConvolutionAwareInitializer from layers, because lr-dropout does the same thing. 2021-07-15 00:50:11 +04:00
iperov 2edac3df8c fix for clipgrad 2021-07-15 00:24:38 +04:00
iperov 35877dbfd7 AMP, SAEHD: added option use_fp16: Increases training/inference speed, reduces model size. Model may crash. Enable it after 1-5k iters.
AMP: removed lr_dropout, now it is enabled by default;
2021-07-14 23:03:54 +04:00
iperov ee1bc83a14 AMP, SAEHD: reverted GAN to december version. 2021-07-07 21:45:43 +04:00
iperov b4b72d056f upd torrent link 2021-07-01 21:58:23 +04:00
iperov 9e0bd34253 update torrent link 2021-07-01 15:50:51 +04:00
iperov 446b95942f fix 2021-07-01 15:08:11 +04:00
iperov 0444a9a565 default morph factor is now 0.1 2021-07-01 14:29:11 +04:00
iperov 075422e0c5 amp: fixes, removed pretrain option 2021-07-01 14:25:54 +04:00
iperov 63c794b3d0 rename dump_dflive to export_dfm 2021-06-28 18:26:54 +04:00
iperov 9d6b6feb1f _ 2021-06-28 14:27:09 +04:00
iperov c43b3b161b _ 2021-06-27 20:43:44 +04:00
iperov 90a74efd89 fix 2021-06-26 23:06:30 +04:00
iperov 714234bce4 missing file 2021-06-26 18:03:04 +04:00
iperov 3c6bbe22b9 AMP: added Inter dimensions param. The model is not changed. Should be equal or more than AutoEncoder dimensions. More dims are better, but require more VRAM. You can fine-tune model size to fit your GPU. 2021-06-26 10:55:33 +04:00
iperov 5783191849 AMP: code refactoring, fix preview history
added dumpdflive command
2021-06-26 10:44:41 +04:00
iperov 6d89d7fa4c _ 2021-06-26 10:42:11 +04:00
iperov 9092e73c41 update QQ group 2021-06-18 17:15:08 +04:00
iperov 427e51b413 update QQ chinesse group 2021-06-18 17:13:29 +04:00
iperov 62f1d57871 AMP: fix a merge bug where morph_factor was always = 1.0 2021-06-14 17:58:01 +04:00
iperov 62c6fffdde fix 2021-06-10 14:42:58 +04:00
iperov 9f0b4bf3cf readme: added tiktok channel 2021-06-09 19:18:35 +04:00
iperov 6e0c8222f7 readme: added tiktok channel 2021-06-09 19:17:33 +04:00
iperov 5dc027a8b0 _ 2021-06-09 19:17:18 +04:00
iperov 24ba84d4a5 MergeMasked: added full mask type as white square 2021-06-06 21:08:42 +04:00
iperov 2a95699b1c upd windows magnet link 2021-06-03 00:00:40 +04:00
iperov 8ea7820b9c fix 2021-06-03 00:00:28 +04:00
iperov a6438ca494 upd magnet link 2021-06-01 13:30:06 +04:00
iperov 5fac5ee4f3 AMP model: added ‘morph_factor’ option. [0.1 .. 0.5]
The smaller the value, the more src-like facial expressions will appear.
The larger the value, the less space there is to train a large dst faceset in the neural network.
Typical fine value is 0.33

AMP model: added ‘pretrain’ mode as in SAEHD
Windows build: Default pretrain dataset is updated with applied Generic XSeg mask
2021-06-01 09:37:55 +04:00
iperov 34b62862e0 fix help msg 2021-06-01 09:36:53 +04:00
iperov 1981ed0ca8 _ 2021-06-01 00:21:27 +04:00
iperov 315f241c51 upd magnet link 2021-05-30 14:28:23 +04:00
iperov e52b53f87c AMP fix 2021-05-30 09:24:23 +04:00
iperov 11a7993238 fix 2021-05-29 22:23:07 +04:00
iperov 6f86c68e65 add AMP model 2021-05-29 18:57:35 +04:00
iperov e6e2ee7466 pixel_norm op 2021-05-25 14:26:48 +04:00
iperov 757283d10e Trainer now saves every 25 min instead of 15 2021-05-25 14:23:25 +04:00
iperov 766750941a Faceset packer now asks whether to delete the original files 2021-05-25 14:21:38 +04:00
iperov 41b517517e XSegEditor , delete button now moves the face to _trash directory and it has been moved to the right border of the window 2021-05-25 14:17:34 +04:00
iperov deedd3dd12 upd windows magnet link 2021-05-12 13:29:01 +04:00
iperov 66bb72f164 XSeg model has been changed to work better with large amount of various faces, thus you should retrain existing xseg model.
Windows build: Added Generic XSeg model pretrained on various faces. It is most suitable for src faceset because it contains clean faces, but also can be applied on dst footage without complex face obstructions.
2021-05-12 13:28:48 +04:00
iperov 65a703c024 SampleGeneratorFaceXSeg: additional sample augmentation for better generalization 2021-05-12 09:42:29 +04:00
iperov 8b90ca0dac XSegUtil apply xseg now checks model face type 2021-05-12 09:41:53 +04:00
iperov b15cdd96a1 fix FacesetResizer 2021-05-09 20:07:36 +04:00
iperov aa26089032 FacesetResizer now supports changing face type 2021-05-06 10:52:04 +04:00
iperov d204e049d1 revert 2021-05-04 23:16:47 +04:00
iperov 6f5bccaa15 XSegEditor: added delete button 2021-04-30 18:58:43 +04:00
iperov 081d8faa45 XSeg sample generator: additional sample augmentation for training 2021-04-30 16:49:32 +04:00
iperov 87030bdcdf revert for more tests 2021-04-27 19:22:11 +04:00
iperov e53d1b1820 XSeg trainer: added random relighting sample augmentation to improve generalization 2021-04-26 10:51:06 +04:00
iperov 23130cd56a fix for colab 2021-04-22 21:12:42 +04:00
iperov 78f12de819 upd windows magnet link 2021-04-22 19:01:26 +04:00
iperov 26b4b6adef upd readme 2021-04-22 18:30:23 +04:00
iperov fdb143ff47 added AMD/Intel cards support via DirectX12 ( DirectML backend ) 2021-04-22 18:19:15 +04:00
iperov fc4a49c3e7 improved xseg sample generator : added random sharpen 2021-04-22 18:18:02 +04:00
iperov dcf146cc16 imagelib : apply_random_sharpen 2021-04-22 18:16:42 +04:00
iperov 7a08c0c1d3 Extractor: extract from dflimg is now the same as from any image 2021-04-20 16:45:20 +04:00
iperov 93fe480eca revert 2021-04-19 13:35:14 +04:00
iperov af0b3904fc XSeg trainer: additional dssim loss. 2021-04-15 21:43:47 +04:00
iperov 65432d0c3d imagelib: random crop func (unused) 2021-04-15 21:43:18 +04:00
iperov bee8628d77 xseg sample generator: additional sample augmentation 2021-04-12 16:52:39 +04:00
iperov d676a365f7 XSeg training generator: added additional sample augmentations 2021-04-10 09:59:49 +04:00
iperov 457a39c093 mathlib update 2021-04-10 09:58:33 +04:00
iperov 243f73fafc core.imagelib: update 2021-04-10 09:58:14 +04:00
iperov d63294a548 fix 'raw-rgb' merge 2021-03-29 09:33:50 +04:00
iperov 30ef9c0808 removing alipay donation 2021-03-28 20:03:38 +04:00
iperov 1652bffeb0 change in xseg apply and fetch 2021-03-24 13:42:57 +04:00
iperov b7245d888b missing file 2021-03-24 13:00:47 +04:00
iperov 0ad9421101 upd windows magnet link 2021-03-23 17:22:42 +04:00
iperov 8d46cd94fd upd pretrain option help 2021-03-23 16:07:50 +04:00
iperov f387179cba Added faceset resize tool via
4.2) data_src util faceset resize.bat
5.2) data_dst util faceset resize.bat

Resize faceset to match model resolution to reduce CPU load during training.

Don’t forget to keep original faceset.
2021-03-23 16:01:24 +04:00
iperov e47b602ec8 SAEHD: random flip replaced with random SRC flip(default False) and random DST flip(default True) 2021-03-23 15:13:10 +04:00
iperov b333fcea4b dump_ckpt 2021-03-23 15:00:24 +04:00
iperov 3d0e18b0ad readme: added 1facerussia tiktok channel 2021-03-12 20:32:14 +04:00
iperov 48181832a7 readme: added deeptomcruise channel 2021-02-26 16:32:45 +04:00
iperov a089c1f108 upd meme 2021-01-27 17:59:00 +04:00