Skip to content

Commit

Permalink
GUI - Plugin preset support
Browse files Browse the repository at this point in the history
  - Add ability to load and save presets for all existing faceswap plugins
  - Add phaze-a presets
  • Loading branch information
torzdf committed May 8, 2021
1 parent eb0cc9c commit cc5bc87
Show file tree
Hide file tree
Showing 16 changed files with 721 additions and 22 deletions.
12 changes: 12 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -43,3 +43,15 @@
!tools/
!tools/**/
!tools/**/*.py

# GUI Plugin Presets
!lib/gui/**/presets/train/model_phaze_a_dfaker_preset.json
!lib/gui/**/presets/train/model_phaze_a_dfl-h128_preset.json
!lib/gui/**/presets/train/model_phaze_a_dfl-sae-df_preset.json
!lib/gui/**/presets/train/model_phaze_a_dfl-sae-liae_preset.json
!lib/gui/**/presets/train/model_phaze_a_dfl-saehd-df_preset.json
!lib/gui/**/presets/train/model_phaze_a_dfl-saehd-liae_preset.json
!lib/gui/**/presets/train/model_phaze_a_iae_preset.json
!lib/gui/**/presets/train/model_phaze_a_lightweight_preset.json
!lib/gui/**/presets/train/model_phaze_a_original_preset.json
!lib/gui/**/presets/train/model_phaze_a_stojo_preset.json
Empty file.
Empty file.
Empty file.
48 changes: 48 additions & 0 deletions lib/gui/.cache/presets/train/model_phaze_a_dfaker_preset.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
{
"output_size": 128,
"shared_fc": "none",
"enable_gblock": false,
"split_fc": false,
"split_gblock": false,
"split_decoders": true,
"enc_architecture": "fs_original",
"enc_scaling": 40,
"enc_load_weights": false,
"bottleneck_type": "dense",
"bottleneck_norm": "none",
"bottleneck_size": 1024,
"bottleneck_in_encoder": true,
"fc_depth": 1,
"fc_min_filters": 1024,
"fc_max_filters": 1024,
"fc_dimensions": 4,
"fc_filter_slope": -0.5,
"fc_dropout": 0.0,
"fc_upsampler": "subpixel",
"fc_upsamples": 1,
"fc_upsample_filters": 512,
"fc_gblock_depth": 3,
"fc_gblock_min_nodes": 512,
"fc_gblock_max_nodes": 512,
"fc_gblock_filter_slope": -0.5,
"fc_gblock_dropout": 0.0,
"dec_upscale_method": "subpixel",
"dec_norm": "none",
"dec_min_filters": 64,
"dec_max_filters": 512,
"dec_filter_slope": -0.45,
"dec_res_blocks": 1,
"dec_output_kernel": 5,
"dec_gaussian": false,
"dec_skip_last_residual": true,
"freeze_layers": "encoder",
"load_layers": "encoder",
"fs_original_depth": 4,
"fs_original_min_filters": 128,
"fs_original_max_filters": 1024,
"mobilenet_width": 1.0,
"mobilenet_depth": 1,
"mobilenet_dropout": 0.001,
"__filetype": "faceswap_preset",
"__section": "train|model|phaze_a"
}
48 changes: 48 additions & 0 deletions lib/gui/.cache/presets/train/model_phaze_a_dfl-h128_preset.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
{
"output_size": 128,
"shared_fc": "none",
"enable_gblock": false,
"split_fc": false,
"split_gblock": false,
"split_decoders": true,
"enc_architecture": "fs_original",
"enc_scaling": 80,
"enc_load_weights": false,
"bottleneck_type": "dense",
"bottleneck_norm": "none",
"bottleneck_size": 512,
"bottleneck_in_encoder": true,
"fc_depth": 1,
"fc_min_filters": 512,
"fc_max_filters": 512,
"fc_dimensions": 8,
"fc_filter_slope": -0.5,
"fc_dropout": 0.0,
"fc_upsampler": "subpixel",
"fc_upsamples": 1,
"fc_upsample_filters": 512,
"fc_gblock_depth": 3,
"fc_gblock_min_nodes": 512,
"fc_gblock_max_nodes": 512,
"fc_gblock_filter_slope": -0.5,
"fc_gblock_dropout": 0.0,
"dec_upscale_method": "subpixel",
"dec_norm": "none",
"dec_min_filters": 128,
"dec_max_filters": 512,
"dec_filter_slope": -0.33,
"dec_res_blocks": 0,
"dec_output_kernel": 5,
"dec_gaussian": false,
"dec_skip_last_residual": false,
"freeze_layers": "encoder",
"load_layers": "encoder",
"fs_original_depth": 4,
"fs_original_min_filters": 128,
"fs_original_max_filters": 1024,
"mobilenet_width": 1.0,
"mobilenet_depth": 1,
"mobilenet_dropout": 0.001,
"__filetype": "faceswap_preset",
"__section": "train|model|phaze_a"
}
48 changes: 48 additions & 0 deletions lib/gui/.cache/presets/train/model_phaze_a_dfl-sae-df_preset.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
{
"output_size": 128,
"shared_fc": "none",
"enable_gblock": false,
"split_fc": false,
"split_gblock": false,
"split_decoders": true,
"enc_architecture": "fs_original",
"enc_scaling": 80,
"enc_load_weights": false,
"bottleneck_type": "dense",
"bottleneck_norm": "none",
"bottleneck_size": 512,
"bottleneck_in_encoder": true,
"fc_depth": 1,
"fc_min_filters": 512,
"fc_max_filters": 512,
"fc_dimensions": 8,
"fc_filter_slope": -0.5,
"fc_dropout": 0.0,
"fc_upsampler": "subpixel",
"fc_upsamples": 1,
"fc_upsample_filters": 512,
"fc_gblock_depth": 3,
"fc_gblock_min_nodes": 512,
"fc_gblock_max_nodes": 512,
"fc_gblock_filter_slope": -0.5,
"fc_gblock_dropout": 0.0,
"dec_upscale_method": "subpixel",
"dec_norm": "none",
"dec_min_filters": 128,
"dec_max_filters": 504,
"dec_filter_slope": -0.33,
"dec_res_blocks": 2,
"dec_output_kernel": 5,
"dec_gaussian": false,
"dec_skip_last_residual": false,
"freeze_layers": "encoder",
"load_layers": "encoder",
"fs_original_depth": 4,
"fs_original_min_filters": 126,
"fs_original_max_filters": 1008,
"mobilenet_width": 1.0,
"mobilenet_depth": 1,
"mobilenet_dropout": 0.001,
"__filetype": "faceswap_preset",
"__section": "train|model|phaze_a"
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
{
"output_size": 128,
"shared_fc": "half",
"enable_gblock": false,
"split_fc": true,
"split_gblock": false,
"split_decoders": false,
"enc_architecture": "fs_original",
"enc_scaling": 80,
"enc_load_weights": false,
"bottleneck_type": "dense",
"bottleneck_norm": "none",
"bottleneck_size": 256,
"bottleneck_in_encoder": false,
"fc_depth": 1,
"fc_min_filters": 512,
"fc_max_filters": 512,
"fc_dimensions": 8,
"fc_filter_slope": -0.5,
"fc_dropout": 0.0,
"fc_upsampler": "subpixel",
"fc_upsamples": 1,
"fc_upsample_filters": 512,
"fc_gblock_depth": 3,
"fc_gblock_min_nodes": 512,
"fc_gblock_max_nodes": 512,
"fc_gblock_filter_slope": -0.5,
"fc_gblock_dropout": 0.0,
"dec_upscale_method": "subpixel",
"dec_norm": "none",
"dec_min_filters": 128,
"dec_max_filters": 504,
"dec_filter_slope": -0.33,
"dec_res_blocks": 2,
"dec_output_kernel": 5,
"dec_gaussian": false,
"dec_skip_last_residual": false,
"freeze_layers": "encoder",
"load_layers": "encoder",
"fs_original_depth": 4,
"fs_original_min_filters": 126,
"fs_original_max_filters": 1008,
"mobilenet_width": 1.0,
"mobilenet_depth": 1,
"mobilenet_dropout": 0.001,
"__filetype": "faceswap_preset",
"__section": "train|model|phaze_a"
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
{
"output_size": 128,
"shared_fc": "none",
"enable_gblock": false,
"split_fc": false,
"split_gblock": false,
"split_decoders": true,
"enc_architecture": "fs_original",
"enc_scaling": 80,
"enc_load_weights": false,
"bottleneck_type": "dense",
"bottleneck_norm": "none",
"bottleneck_size": 256,
"bottleneck_in_encoder": false,
"fc_depth": 1,
"fc_min_filters": 256,
"fc_max_filters": 256,
"fc_dimensions": 8,
"fc_filter_slope": -0.5,
"fc_dropout": 0.0,
"fc_upsampler": "subpixel",
"fc_upsamples": 1,
"fc_upsample_filters": 256,
"fc_gblock_depth": 3,
"fc_gblock_min_nodes": 512,
"fc_gblock_max_nodes": 512,
"fc_gblock_filter_slope": -0.5,
"fc_gblock_dropout": 0.0,
"dec_upscale_method": "subpixel",
"dec_norm": "none",
"dec_min_filters": 128,
"dec_max_filters": 512,
"dec_filter_slope": -0.33,
"dec_res_blocks": 1,
"dec_output_kernel": 1,
"dec_gaussian": false,
"dec_skip_last_residual": false,
"freeze_layers": "encoder",
"load_layers": "encoder",
"fs_original_depth": 4,
"fs_original_min_filters": 64,
"fs_original_max_filters": 5124,
"mobilenet_width": 1.0,
"mobilenet_depth": 1,
"mobilenet_dropout": 0.001,
"__filetype": "faceswap_preset",
"__section": "train|model|phaze_a"
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
{
"output_size": 128,
"shared_fc": "half",
"enable_gblock": false,
"split_fc": true,
"split_gblock": false,
"split_decoders": false,
"enc_architecture": "fs_original",
"enc_scaling": 80,
"enc_load_weights": false,
"bottleneck_type": "dense",
"bottleneck_norm": "none",
"bottleneck_size": 256,
"bottleneck_in_encoder": false,
"fc_depth": 1,
"fc_min_filters": 512,
"fc_max_filters": 512,
"fc_dimensions": 8,
"fc_filter_slope": -0.5,
"fc_dropout": 0.0,
"fc_upsampler": "subpixel",
"fc_upsamples": 1,
"fc_upsample_filters": 512,
"fc_gblock_depth": 3,
"fc_gblock_min_nodes": 512,
"fc_gblock_max_nodes": 512,
"fc_gblock_filter_slope": -0.5,
"fc_gblock_dropout": 0.0,
"dec_upscale_method": "subpixel",
"dec_norm": "none",
"dec_min_filters": 128,
"dec_max_filters": 512,
"dec_filter_slope": -0.33,
"dec_res_blocks": 1,
"dec_output_kernel": 1,
"dec_gaussian": false,
"dec_skip_last_residual": false,
"freeze_layers": "encoder",
"load_layers": "encoder",
"fs_original_depth": 4,
"fs_original_min_filters": 64,
"fs_original_max_filters": 512,
"mobilenet_width": 1.0,
"mobilenet_depth": 1,
"mobilenet_dropout": 0.001,
"__filetype": "faceswap_preset",
"__section": "train|model|phaze_a"
}
48 changes: 48 additions & 0 deletions lib/gui/.cache/presets/train/model_phaze_a_iae_preset.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
{
"output_size": 64,
"shared_fc": "full",
"enable_gblock": false,
"split_fc": true,
"split_gblock": false,
"split_decoders": false,
"enc_architecture": "fs_original",
"enc_scaling": 40,
"enc_load_weights": false,
"bottleneck_type": "dense",
"bottleneck_norm": "none",
"bottleneck_size": 1024,
"bottleneck_in_encoder": false,
"fc_depth": 1,
"fc_min_filters": 512,
"fc_max_filters": 512,
"fc_dimensions": 4,
"fc_filter_slope": -0.5,
"fc_dropout": 0.0,
"fc_upsampler": "subpixel",
"fc_upsamples": 0,
"fc_upsample_filters": 512,
"fc_gblock_depth": 3,
"fc_gblock_min_nodes": 512,
"fc_gblock_max_nodes": 512,
"fc_gblock_filter_slope": -0.5,
"fc_gblock_dropout": 0.0,
"dec_upscale_method": "subpixel",
"dec_norm": "none",
"dec_min_filters": 64,
"dec_max_filters": 512,
"dec_filter_slope": -0.45,
"dec_res_blocks": 0,
"dec_output_kernel": 5,
"dec_gaussian": false,
"dec_skip_last_residual": false,
"freeze_layers": "encoder",
"load_layers": "encoder",
"fs_original_depth": 4,
"fs_original_min_filters": 128,
"fs_original_max_filters": 1024,
"mobilenet_width": 1.0,
"mobilenet_depth": 1,
"mobilenet_dropout": 0.001,
"__filetype": "faceswap_preset",
"__section": "train|model|phaze_a"
}
Loading

0 comments on commit cc5bc87

Please sign in to comment.