filmoswop писал(а):
82764153Разработчику спасибо за интересный инструмент.
Пара соображений по работе с прогой, может кому пригодится. Актуально для GTX и RTX с 6-8 гигабайтами памяти. 12 гб видеопамяти для проекта пока не пробовал, но в будущем очень хочу.
Разрешение модели - 160. Число dims не меняем, так вписывается модель в наш объем памяти и использует ее почти на максимуме (сам максимум где-то в коде зашит, и он несколько не достигает всех возможностей видюхи). Хорошо тренированная модель (для src от 10 миллионов и выше, dst различные) около 1500-2000 кадров нормально считается сутки.
При сборке, для моделей df - неплохо erode mask -35 и blur mask 210. Для модели liae: -15 и 100 соответственно.
Color Transfer для всех практически случаев - неплохо MKL справляется. Super Resolution включаю на 50.
Для улучшения как кадров dst так и кадров src несколько раз успешно использовал проект с гитхаба GFPGAN, неожиданно классно. Два примера из жизни. Старое кино с нечеткими лицами и низким разрешением, в части кадров лица не распознались. Прогоняем все кадры dst за пару часов через GFPGAN, он улучшает лица, далее они все распознаются, кадров с пропущенными лицами нет. Второй пример: часть лиц модели src не очень высокого качества, для тренировки лучше подсовывать самые четкие лица. Ок, сортируем по четкости, все лица плохого качества отбираем и прогоняем через GFPGAN - получаем идеальную модель с лицами только высокой четкости.
1080ти - но на 10ке врет про 9 гигов (за GFPGAN спасибо)
==================== Model Summary ====================
== ==
== Model name: new_SAEHD ==
== ==
== Current iteration: 3971758 ==
== ==
==------------------ Model Options ------------------==
== ==
== resolution: 224 ==
== face_type: wf ==
== models_opt_on_gpu: True ==
== archi: liae-ud ==
== ae_dims: 256 ==
== e_dims: 64 ==
== d_dims: 64 ==
== d_mask_dims: 22 ==
== masked_training: True ==
== uniform_yaw: True ==
== adabelief: True ==
== lr_dropout: n ==
== random_warp: False ==
== gan_power: 0.01 ==
== true_face_power: 0.0 ==
== face_style_power: 0.0 ==
== bg_style_power: 0.0 ==
== ct_mode: none ==
== clipgrad: True ==
== pretrain: False ==
== autobackup_hour: 0 ==
== write_preview_history: False ==
== target_iter: 0 ==
== random_flip: False ==
== batch_size: 18 ==
== gan_warmup: 0.0 ==
== eyes_mouth_prio: False ==
== gan_patch_size: 28 ==
== gan_dims: 32 ==
== random_src_flip: False ==
== random_dst_flip: False ==
== ==
==------------------- Running On --------------------==
== ==
== Device index: 0 ==
== Name: NVIDIA GeForce GTX 1080 Ti ==
== VRAM: 9.42GB ==
== ==
=======================================================
