80cae7d9f5
_
2021-10-11 15:03:34 +04:00
f8469fe4d7
fix sample processor
2021-10-09 13:56:53 +04:00
3aa2b56eda
fix
2021-09-30 00:40:59 +04:00
c6d0c547b7
update optimizers
2021-09-29 22:06:46 +04:00
9e0079c6a0
SAEHD, AMP: removed the implicit function of periodically retraining last 16 “high-loss” samples
2021-09-29 16:48:54 +04:00
33ff0be722
_
2021-09-29 16:41:43 +04:00
6e094d873d
SAEHD: removed 'dst_denoise' option. Added -t arhi option.
2021-08-29 11:48:55 +04:00
b256b07e03
fix gaussian_blur
2021-08-20 17:06:11 +04:00
57f3393ec2
tf add unused TanhPolar
2021-08-20 17:05:26 +04:00
56e70edc46
fix error in model saving
2021-08-19 23:18:04 +04:00
4aa33d3ba8
_
2021-08-12 14:37:57 +04:00
a299324166
_
2021-08-12 14:28:36 +04:00
4581d48ff0
_
2021-08-10 20:54:54 +04:00
55b947eab5
XSeg: added pretrain option.
2021-07-30 17:24:21 +04:00
f044c99ddc
remove ConvolutionAwareInitializer from layers, because lr-dropout does the same thing.
2021-07-15 00:50:11 +04:00
2edac3df8c
fix for clipgrad
2021-07-15 00:24:38 +04:00
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
ee1bc83a14
AMP, SAEHD: reverted GAN to december version.
2021-07-07 21:45:43 +04:00
62c6fffdde
fix
2021-06-10 14:42:58 +04:00
5dc027a8b0
_
2021-06-09 19:17:18 +04:00
e52b53f87c
AMP fix
2021-05-30 09:24:23 +04:00
e6e2ee7466
pixel_norm op
2021-05-25 14:26:48 +04:00
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
081d8faa45
XSeg sample generator: additional sample augmentation for training
2021-04-30 16:49:32 +04:00
e53d1b1820
XSeg trainer: added random relighting sample augmentation to improve generalization
2021-04-26 10:51:06 +04:00
23130cd56a
fix for colab
2021-04-22 21:12:42 +04:00
fdb143ff47
added AMD/Intel cards support via DirectX12 ( DirectML backend )
2021-04-22 18:19:15 +04:00
dcf146cc16
imagelib : apply_random_sharpen
2021-04-22 18:16:42 +04:00
65432d0c3d
imagelib: random crop func (unused)
2021-04-15 21:43:18 +04:00
457a39c093
mathlib update
2021-04-10 09:58:33 +04:00
243f73fafc
core.imagelib: update
2021-04-10 09:58:14 +04:00
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
11add4cd4f
update GAN model
2021-01-03 19:25:39 +04:00
54fc3162ed
new gan
2021-01-03 01:08:20 +04:00
140f16f772
Merge branch 'master' into new_gan
2021-01-01 18:04:10 +04:00
4f2efd7985
fix support for v1/v2
2021-01-01 17:59:57 +04:00
8ff34be5e4
leras.nn : support for tf ver 1
2021-01-01 17:37:12 +04:00
ad5733c5bb
upd PatchDiscriminator
2020-12-31 17:38:41 +04:00
ae9e16b4a5
SAEHD: new GAN
2020-12-30 14:33:14 +04:00
241d1a9c35
leras ops: pixelnorm, total_variation_mse
2020-12-30 14:32:07 +04:00
299d91c81e
remove unused
2020-12-30 14:31:23 +04:00
cc950f12fb
set CUDA_CACHE_MAXSIZE to 2gb before first cuda load
2020-12-28 18:37:29 +04:00
b7bed0ef5e
optimize MPSharedList.py
2020-12-20 23:04:00 +04:00
977d8a2d77
The load time of training data has been reduced significantly.
2020-12-20 19:17:24 +04:00
e7d36b4287
leras: added AdaBelief optimizer
2020-12-11 15:47:47 +04:00
35945b257c
fix dimensions error
2020-12-11 14:34:37 +04:00
b9c9e7cffd
fix depth_to_space for tf2.4.0. Removing compute_output_shape in leras, because it uses CPU device, which does not support all ops.
2020-12-11 11:28:33 +04:00
bb432b21f9
.
2020-11-25 12:19:43 +04:00
c516454566
fix rmsprop
2020-11-18 13:59:14 +04:00
874a7eba18
1
2020-11-18 13:41:19 +04:00
iperov