| """ | |
| 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' | |
| } |