Commit graph

164 commits

Author SHA1 Message Date
iperov
521a23f557 fix scale factor in sample generation. added util --export-faceset-mask 2022-07-23 10:30:22 +04:00
iperov
0d19d8ec8e fix for tf-directml 2022-05-04 16:41:17 +04:00
iperov
7326771c02 fixed rct color transfer function, incorrectly clip colors before 2021-10-23 09:53:40 +04:00
iperov
14cc9d4e5f _ 2021-10-17 21:37:22 +04:00
iperov
80cae7d9f5 _ 2021-10-11 15:03:34 +04:00
iperov
f8469fe4d7 fix sample processor 2021-10-09 13:56:53 +04:00
iperov
3aa2b56eda fix 2021-09-30 00:40:59 +04:00
iperov
c6d0c547b7 update optimizers 2021-09-29 22:06:46 +04:00
iperov
9e0079c6a0 SAEHD, AMP: removed the implicit function of periodically retraining last 16 “high-loss” samples 2021-09-29 16:48:54 +04:00
iperov
33ff0be722 _ 2021-09-29 16:41:43 +04:00
iperov
6e094d873d SAEHD: removed 'dst_denoise' option. Added -t arhi option. 2021-08-29 11:48:55 +04:00
iperov
b256b07e03 fix gaussian_blur 2021-08-20 17:06:11 +04:00
iperov
57f3393ec2 tf add unused TanhPolar 2021-08-20 17:05:26 +04:00
iperov
56e70edc46 fix error in model saving 2021-08-19 23:18:04 +04:00
iperov
4aa33d3ba8 _ 2021-08-12 14:37:57 +04:00
iperov
a299324166 _ 2021-08-12 14:28:36 +04:00
iperov
4581d48ff0 _ 2021-08-10 20:54:54 +04:00
iperov
55b947eab5 XSeg: added pretrain option. 2021-07-30 17:24:21 +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
62c6fffdde fix 2021-06-10 14:42:58 +04:00
iperov
5dc027a8b0 _ 2021-06-09 19:17:18 +04:00
iperov
e52b53f87c AMP fix 2021-05-30 09:24:23 +04:00
iperov
e6e2ee7466 pixel_norm op 2021-05-25 14:26:48 +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
081d8faa45 XSeg sample generator: additional sample augmentation for training 2021-04-30 16:49:32 +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
fdb143ff47 added AMD/Intel cards support via DirectX12 ( DirectML backend ) 2021-04-22 18:19:15 +04:00
iperov
dcf146cc16 imagelib : apply_random_sharpen 2021-04-22 18:16:42 +04:00
iperov
65432d0c3d imagelib: random crop func (unused) 2021-04-15 21:43:18 +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
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
11add4cd4f update GAN model 2021-01-03 19:25:39 +04:00
iperov
54fc3162ed new gan 2021-01-03 01:08:20 +04:00
iperov
140f16f772 Merge branch 'master' into new_gan 2021-01-01 18:04:10 +04:00
iperov
4f2efd7985 fix support for v1/v2 2021-01-01 17:59:57 +04:00
iperov
8ff34be5e4 leras.nn : support for tf ver 1 2021-01-01 17:37:12 +04:00
iperov
ad5733c5bb upd PatchDiscriminator 2020-12-31 17:38:41 +04:00
iperov
ae9e16b4a5 SAEHD: new GAN 2020-12-30 14:33:14 +04:00
iperov
241d1a9c35 leras ops: pixelnorm, total_variation_mse 2020-12-30 14:32:07 +04:00
iperov
299d91c81e remove unused 2020-12-30 14:31:23 +04:00
iperov
cc950f12fb set CUDA_​CACHE_​MAXSIZE to 2gb before first cuda load 2020-12-28 18:37:29 +04:00
iperov
b7bed0ef5e optimize MPSharedList.py 2020-12-20 23:04:00 +04:00
iperov
977d8a2d77 The load time of training data has been reduced significantly. 2020-12-20 19:17:24 +04:00
iperov
e7d36b4287 leras: added AdaBelief optimizer 2020-12-11 15:47:47 +04:00
iperov
35945b257c fix dimensions error 2020-12-11 14:34:37 +04:00