config.py

"""
Configuration constants for HATSAT application.
"""

# Model configuration
MODEL_CHECKPOINT = 'net_g_150000.pth'
REQUIRED_IMAGE_SIZE = (130, 130)
WINDOW_SIZE = 16
UPSCALE_FACTOR = 4

# Model architecture parameters
MODEL_CONFIG = {
    'upscale': 4,
    'in_chans': 3,
    'img_size': 128,
    'window_size': 16,
    'compress_ratio': 3,
    'squeeze_factor': 30,
    'conv_scale': 0.01,
    'overlap_ratio': 0.5,
    'img_range': 1.,
    'depths': [6, 6, 6, 6, 6, 6],
    'embed_dim': 180,
    'num_heads': [6, 6, 6, 6, 6, 6],
    'mlp_ratio': 2,
    'upsampler': 'pixelshuffle',
    'resi_connection': '1conv'
}