allenai / allenai/satlaspretrain_models
Sentinel-2 ResNet50 multi-image MS model does not accept multi-image input
- Langage dominant
- Jupyter Notebook
- Étoiles
- 155
- Forks
- 23
- Métriques de merge des PR
- Aucune PR mergée en 30 j
Description
The Sentinel2_Resnet50_MI_MS (multi-image) model is exactly the same as the Sentinel2_Resnet50_SI_MS (single image) model, and hence does not accept multi-image input.
In model.py, the following code means that a ResNet model is never loaded as a multi-image model. If the first if statement is commented out, then there is an error in loading the state_dict for Sentinel2_Resnet50_MI_MS.
```
# If using a model for multi-image, need the Aggretation to wrap underlying backbone model.
prefix, prefix_allowed_count = None, None
if backbone_arch in [Backbone.RESNET50, Backbone.RESNET152]:
prefix_allowed_count = 0
elif multi_image:
backbone = AggregationBackbone(num_channels, backbone)
prefix_allowed_count = 2
else:
prefix_allowed_count = 1
```
Guide de contribution
Aucun guide de contribution indexé pour ce dépôt
Évaluation
Cette issue n'a pas encore été évaluée.