From 7f6c65fa954a43c9d248219525cad35fe1b8a046 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 6 Apr 2021 21:19:26 +0800 Subject: Network modification This commit change the auto-encoder by removing fc and optimizing latent space features --- config.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'config.py') diff --git a/config.py b/config.py index 66eab98..04f7b36 100644 --- a/config.py +++ b/config.py @@ -11,7 +11,7 @@ config: Configuration = { # Recorde disentangled image or not 'image_log_on': False, # The number of subjects for validating (Part of testing set) - 'val_size': 10, + 'val_size': 20, }, # Dataset settings 'dataset': { @@ -51,7 +51,7 @@ config: Configuration = { # Auto-encoder feature channels coefficient 'ae_feature_channels': 64, # Appearance, canonical and pose feature dimensions - 'f_a_c_p_dims': (192, 192, 96), + 'f_a_c_p_dims': (192, 192, 128), # HPM pyramid scales, of which sum is number of parts 'hpm_scales': (1, 2, 4, 8), # Global pooling method -- cgit v1.2.3