Update model.py
Browse files
model.py
CHANGED
|
@@ -10,7 +10,7 @@ class UNet(nn.Module):
|
|
| 10 |
|
| 11 |
def __init__(self,
|
| 12 |
encoder_name='tu-regnetz_d8',
|
| 13 |
-
encoder_weights=
|
| 14 |
in_channels=13, # Number of input channels (13 for Sentinel-2 multi-spectral images)
|
| 15 |
num_classes=4, # Number of output classes (e.g., clear, thick cloud, thin cloud, cloud shadow)
|
| 16 |
freeze_encoder=False): # Whether to freeze the encoder's weights
|
|
|
|
| 10 |
|
| 11 |
def __init__(self,
|
| 12 |
encoder_name='tu-regnetz_d8',
|
| 13 |
+
encoder_weights=None,
|
| 14 |
in_channels=13, # Number of input channels (13 for Sentinel-2 multi-spectral images)
|
| 15 |
num_classes=4, # Number of output classes (e.g., clear, thick cloud, thin cloud, cloud shadow)
|
| 16 |
freeze_encoder=False): # Whether to freeze the encoder's weights
|