Commit graph

220 commits

Author SHA1 Message Date
iperov
0c1638e168 fix strange opencv behaviour with border_transparent 2019-02-07 11:14:41 +04:00
iperov
5a1fb199f4 added support of non-english characters in paths 2019-02-05 19:33:55 +04:00
iperov
c3b6330284 SAE: now asking face_type after resolution 2019-02-05 15:00:27 +04:00
iperov
883da4b9ed fix Util.py, SAE: lowest ae dims now can be 32 (for testing) 2019-02-05 14:43:54 +04:00
iperov
241ae83e0b upd readme 2019-02-05 13:27:18 +04:00
iperov
707313eb47 SAE+ConverterMasked now offer default values such as mode, erode, blur, values based on model settings. 2019-02-03 21:44:30 +04:00
iperov
e226ab5385 SAE: increased speed of training by 10-18%,
increased clipping border mask in full face mode results better transition of cheeks,
default archi now 'df'
2019-02-03 20:32:46 +04:00
iperov
541d6e1cc9 nothing interesting 2019-01-26 06:50:52 +04:00
iperov
e5b14672ee SAE: upd help for new 'pixel loss' option 2019-01-25 10:02:30 +04:00
iperov
ef3cf392c5 SAE: added pixel loss option. 2019-01-25 09:56:40 +04:00
iperov
1e7a0836f7 nothing interesting 2019-01-24 21:28:54 +04:00
iperov
7f4d261a99 change SAE help message 2019-01-22 18:31:00 +04:00
iperov
6d7c56309d SAE: removed adaptive kernel size. No backward compatibility. 2019-01-22 13:30:38 +04:00
iperov
b6c4171ea1 optimizations of nnlib and SampleGeneratorFace,
refactorings
2019-01-22 11:52:04 +04:00
iperov
fe66b6b2a1 SAE: added 'learn mask' option, default:Y,
SAE: added clipping mask at borders to remove artifact lines.
2019-01-19 16:15:49 +04:00
iperov
dd10d963d1 SAE: upd default params, upd readme 2019-01-19 08:58:25 +04:00
iperov
946688567d enhanced SAE model. You should to restart training.
new default style power = 2.0
fix DF default batch sizes.
upd readme
2019-01-17 21:41:40 +04:00
iperov
22401cecfc fix H128 2019-01-15 17:52:54 +04:00
iperov
ba06a71fff SAE: changed default style power to 10.0 . Now style power is floating number in valid range 0.0 to 100.0 2019-01-14 21:24:33 +04:00
iperov
1f2b1481ef now you can train models on multiple GPU's on same workspace without cloning any folders.
Model files names will be prefixed with GPU index if GPU choosed explicitly on train/convert start.
if you leave GPU idx choice default, then best GPU idx will be choosed and model file names will not contain index prefix.
It gives you possibility to train same fake with various models or options on multiple GPUs.

H64 and H128: now you can choose 'Lighter autoencoder'. It is same as vram gb <= 4 before this update.

added archived_models.zip contains old experiments

RecycleGAN: archived

devicelib: if your system has no NVML installed (some old cards), then it will work with gpu_idx=0 as 'Generic GeForce GPU' with 2GB vram.

refactorings
2019-01-14 10:48:23 +04:00
iperov
b68612e6cd SAE: 'adaptive kernel size' now as an option 2019-01-13 10:03:01 +04:00
iperov
46aa61050a improvement SAE model. New option added. You should to restart training. 2019-01-12 10:10:50 +04:00
iperov
2473dabf37 fix loss history 2019-01-11 19:19:29 +04:00
iperov
e6fa021c3d fix prompt override model settings in convert mode 2019-01-11 11:29:18 +04:00
iperov
32f0db3dea fix and upd readme 2019-01-11 06:19:15 +04:00
iperov
e39d1a2619 fixes 2019-01-10 23:01:33 +04:00
iperov
bd2b575718 fix ModelBase.py 2019-01-10 19:38:31 +04:00
iperov
f82bc7031e fix SAE 2019-01-10 19:29:17 +04:00
iperov
bf5282f7ec if you have multi GPU in a system, you can choose what GPU to use on model start 2019-01-10 19:22:54 +04:00
iperov
48d0123f0b SAE: new options face_style_power, bg_style_power instead of style_power. Zero - means dont use style.
SAE: new option 'lighter_encoder'.
Now model settings can be overrided by pressing enter in 2 seconds while model loading (works on Windows).
Removed all MultiGPU models, because keras multi_gpu in fact doesn't work.
2019-01-10 18:28:12 +04:00
iperov
1e28ae24b1 Fix bug in ConverterMasked when overexposed pixels appear sometimes 2019-01-09 13:57:26 +04:00
iperov
faa8cebc98 fix 'raw' mode in ConverterMasked 2019-01-09 13:40:33 +04:00
iperov
42f0e438f3 added 'raw' mode to convertermasked 2019-01-09 13:35:10 +04:00
iperov
1c601dce4a fix converter bug when blur size == 0 2019-01-09 12:10:29 +04:00
iperov
00d7a3ab4d sae: removing mask in preview window 2019-01-09 11:24:58 +04:00
iperov
e300662f89 SAE: added options archi, learn_face_style , learn_bg_style, style_power 2019-01-09 10:51:31 +04:00
iperov
0c3559d2dd fix sae 2019-01-09 10:46:12 +04:00
iperov
1a2555e160 added '?' help for model options. Added 'Src face scale modifier' to model options. 2019-01-09 09:22:22 +04:00
iperov
e8620919a7 update devicelib/nnlib to detect compute capability 2019-01-08 10:48:06 +04:00
iperov
75f7a9575a SAE: changed base erode/blur mask size for converter 2019-01-07 17:46:50 +04:00
iperov
6faabcb94e DF: fix default mask 2019-01-07 10:13:50 +04:00
iperov
3b0b1a7dec removed UFM model,
added 'random_flip' option to all models, by default - true
2019-01-06 19:59:53 +04:00
iperov
8e1d337e00 added new SAE model - Styled AutoEncoder 2019-01-05 09:57:49 +04:00
iperov
08550ac856 fix H128 after refactoring 2019-01-04 18:28:07 +04:00
iperov
69d5b1e662 fix UFM 2019-01-03 23:02:38 +04:00
iperov
7b70e7eec1 added new model U-net Face Morpher.
removed AVATAR - useless model was just for demo
removed MIAEF128 - use UFM insted
removed LIAEF128YAW - use model option sort by yaw on start for any model
All models now ask some options on start.
Session options (such as target epoch, batch_size, write_preview_history etc) can be overrided by special command arg.
Converter now always ask options and no more support to define options via command line.
fix bug when ConverterMasked always used not predicted mask.
SampleGenerator now always generate samples with replicated border, exclude mask samples.
refactorings
2019-01-02 17:26:12 +04:00
iperov
46bda1d683 RecycleGAN : set minimum bs to 10 2018-12-28 23:42:05 +04:00
iperov
474fff248f RecycleGAN fixes 2018-12-28 23:13:29 +04:00
iperov
f8824f9601 refactoring. Added RecycleGAN for testing. 2018-12-28 19:38:52 +04:00
iperov
8a223845fb refactoring 2018-12-24 13:45:40 +04:00