diff options
author | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-21 19:01:20 +0800 |
---|---|---|
committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-21 19:01:20 +0800 |
commit | 1014543e8b2cecbe6fdf1a0135bbefccee9c0d41 (patch) | |
tree | 90d726834bec29c8816800c71b75aad663edc858 /config.py | |
parent | c538919cb69e35a46811aef0b23baefe6a4c499c (diff) | |
parent | c52fdc2748e272a5195303299a9739291be32281 (diff) |
Merge branch 'master' into python3.8
# Conflicts:
# models/part_net.py
# models/rgb_part_net.py
Diffstat (limited to 'config.py')
-rw-r--r-- | config.py | 12 |
1 files changed, 2 insertions, 10 deletions
@@ -49,22 +49,14 @@ config: Configuration = { # Auto-encoder feature channels coefficient 'ae_feature_channels': 64, # Appearance, canonical and pose feature dimensions - 'f_a_c_p_dims': (128, 128, 64), + 'f_a_c_p_dims': (192, 192, 96), # Use 1x1 convolution in dimensionality reduction 'hpm_use_1x1conv': False, # HPM pyramid scales, of which sum is number of parts 'hpm_scales': (1, 2, 4), # Global pooling method 'hpm_use_avg_pool': True, - 'hpm_use_max_pool': False, - # FConv feature channels coefficient - 'fpfe_feature_channels': 32, - # FConv blocks kernel sizes - 'fpfe_kernel_sizes': ((5, 3), (3, 3), (3, 3)), - # FConv blocks paddings - 'fpfe_paddings': ((2, 1), (1, 1), (1, 1)), - # FConv blocks halving - 'fpfe_halving': (0, 2, 3), + 'hpm_use_max_pool': True, # Attention squeeze ratio 'tfa_squeeze_ratio': 4, # Number of parts after Part Net |