Upload 8 files

.gitattributes

@@ -38,3 +38,9 @@ NonReference_RGBN_x4/hard_constraint.safetensor filter=lfs diff=lfs merge=lfs -t
 NonReference_RGBN_x4/model.safetensor filter=lfs diff=lfs merge=lfs -text
 Reference_RSWIR_x2/example_data.safetensor filter=lfs diff=lfs merge=lfs -text
 Reference_RSWIR_x2/model.safetensor filter=lfs diff=lfs merge=lfs -text
+main/example_data.safetensor filter=lfs diff=lfs merge=lfs -text
+main/f2_model.safetensor filter=lfs diff=lfs merge=lfs -text
+main/hard_constraint.safetensor filter=lfs diff=lfs merge=lfs -text
+main/model.safetensor filter=lfs diff=lfs merge=lfs -text
+main/sr_hard_constraint.safetensor filter=lfs diff=lfs merge=lfs -text
+main/sr_model.safetensor filter=lfs diff=lfs merge=lfs -text

main/example_data.safetensor

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:34d594d36a1220c526a55ce086af5f3f6edfd73f4dd966464bfc8b08220854ac
+size 655440

main/f2_model.safetensor

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1d1b93248ca0bb24cc20a6c2e83b2acd70517357fa322f9051af8f378d54eecc
+size 137790600

main/hard_constraint.safetensor

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cb6370116a248e6d58728a7ec3c3df11fdeae0f1c82ec0e3b424a3734e3f3ba9
+size 1048656

main/mlm.json

@@ -0,0 +1,215 @@
+{
+    "type": "Feature",
+    "stac_version": "1.1.0",
+    "stac_extensions": [
+        "https://stac-extensions.github.io/mlm/v1.4.0/schema.json"
+    ],
+    "id": "Swin2SR model",
+    "geometry": {
+        "type": "Polygon",
+        "coordinates": [
+            [
+                [
+                    -180.0,
+                    -90.0
+                ],
+                [
+                    -180.0,
+                    90.0
+                ],
+                [
+                    180.0,
+                    90.0
+                ],
+                [
+                    180.0,
+                    -90.0
+                ],
+                [
+                    -180.0,
+                    -90.0
+                ]
+            ]
+        ]
+    },
+    "bbox": [
+        -180,
+        -90,
+        180,
+        90
+    ],
+    "properties": {
+        "start_datetime": "1900-01-01T00:00:00Z",
+        "end_datetime": "9999-01-01T00:00:00Z",
+        "description": "A Swin2SR model trained on the CloudSEN12+ dataset.",
+        "dependencies": [
+            "torch",
+            "safetensors.torch",
+            "timm",
+            "mamba_ssm",
+            "einops",
+            "sen2sr"
+        ],
+        "mlm:framework": "pytorch",
+        "mlm:framework_version": "2.1.2+cu121",
+        "file:size": 40455416,
+        "mlm:memory_size": 1,
+        "mlm:accelerator": "cuda",
+        "mlm:accelerator_constrained": false,
+        "mlm:accelerator_summary": "Unknown",
+        "mlm:name": "SR_S2_BestModel",
+        "mlm:architecture": "Swin2SR",
+        "mlm:tasks": [
+            "super-resolution"
+        ],
+        "mlm:input": [
+            {
+                "name": "Sentinel-2 10m and 20m bands (10 Bands)",
+                "bands": [
+                    "B02",
+                    "B03",
+                    "B04",
+                    "B05",
+                    "B06",
+                    "B07",
+                    "B08",
+                    "B8A",
+                    "B11",
+                    "B12"
+                ],
+                "input": {
+                    "shape": [
+                        -1,
+                        10,
+                        128,
+                        128
+                    ],
+                    "dim_order": [
+                        "batch",
+                        "channel",
+                        "height",
+                        "width"
+                    ],
+                    "data_type": "float16"
+                },
+                "pre_processing_function": null
+            }
+        ],
+        "mlm:output": [
+            {
+                "name": "super-resolution",
+                "tasks": [
+                    "super-resolution"
+                ],
+                "result": {
+                    "shape": [
+                        -1,
+                        10,
+                        512,
+                        512
+                    ],
+                    "dim_order": [
+                        "batch",
+                        "channel",
+                        "height",
+                        "width"
+                    ],
+                    "data_type": "float16"
+                },
+                "classification:classes": [],
+                "post_processing_function": null
+            }
+        ],
+        "mlm:total_parameters": 12894526,
+        "mlm:pretrained": true,
+        "datetime": null
+    },
+    "links": [],
+    "assets": {
+        "trainable": {
+            "href": "https://huggingface.co/tacofoundation/SEN2SR/resolve/main/SEN2SR/main/model.safetensor",
+            "type": "application/octet-stream; application=safetensor",
+            "title": "Pytorch model weights checkpoint",
+            "description": "The weights of the model in safetensor format.",
+            "mlm:artifact_type": "safetensor.torch.save_file",
+            "roles": [
+                "mlm:model",
+                "mlm:weights",
+                "data"
+            ]
+        },
+        "hardconstraint": {
+            "href": "https://huggingface.co/tacofoundation/SEN2SR/resolve/main/SEN2SR/main/hard_constraint.safetensor",
+            "type": "application/octet-stream; application=safetensor",
+            "title": "Pytorch hard constraint weights checkpoint. It is used to load faster the hard constraint module.",
+            "description": "The weights of the model in safetensor format.",
+            "mlm:artifact_type": "safetensor.torch.save_file",
+            "roles": [
+                "mlm:model",
+                "mlm:weights",
+                "data"
+            ]
+        },
+        "source_code": {
+            "href": "https://huggingface.co/tacofoundation/SEN2SR/resolve/main/SEN2SR/main/load.py",
+            "type": "text/x-python",
+            "title": "Model load script",
+            "description": "Python script to load the model.",
+            "roles": [
+                "mlm:source_code",
+                "code"
+            ]
+        },
+        "example_data": {
+            "href": "https://huggingface.co/tacofoundation/SEN2SR/resolve/main/SEN2SR/main/example_data.safetensor",
+            "type": "application/octet-stream; application=safetensors",
+            "title": "Example Sentinel-2 image",
+            "description": "Example Sentinel-2 image for model inference.",
+            "roles": [
+                "mlm:example_data",
+                "data"
+            ]
+        },
+        "sr_model": {
+            "href": "https://huggingface.co/tacofoundation/SEN2SR/resolve/main/SEN2SR/main/sr_model.safetensor",
+            "type": "application/octet-stream; application=safetensors",
+            "title": "Auxiliar RGBN super-resolution model",
+            "description": "Auxiliar RGBN super-resolution model used to convert RGBN band to 2.5m resolution.",
+            "roles": [
+                "mlm:weights",
+                "data"
+            ]
+        },
+        "sr_hard_constraint": {
+            "href": "https://huggingface.co/tacofoundation/SEN2SR/resolve/main/SEN2SR/main/sr_hard_constraint.safetensor",
+            "type": "application/octet-stream; application=safetensors",
+            "title": "Auxiliar RGBN super-resolution hard constraint",
+            "description": "Auxiliar RGBN super-resolution hard constraint used to convert RGBN band to 2.5m resolution.",
+            "roles": [
+                "mlm:weights",
+                "data"
+            ]
+        },
+        "f2_model": {
+            "href": "https://huggingface.co/tacofoundation/SEN2SR/resolve/main/SEN2SR/main/f2_model.safetensor",
+            "type": "application/octet-stream; application=safetensors",
+            "title": "Auxiliar RSWIRs super-resolution model",
+            "description": "Auxiliar RSWIRs super-resolution model used to convert RSWIRs band to 10m resolution.",
+            "roles": [
+                "mlm:weights",
+                "data"
+            ]
+        },
+        "f2_hard_constraint": {
+            "href": "https://huggingface.co/tacofoundation/SEN2SR/resolve/main/SEN2SR/main/f2_hard_constraint.safetensor",
+            "type": "application/octet-stream; application=safetensors",
+            "title": "Auxiliar RSWIRs super-resolution hard constraint",
+            "description": "Auxiliar RSWIRs super-resolution hard constraint used to convert RSWIRs band to 10m resolution.",
+            "roles": [
+                "mlm:weights",
+                "data"
+            ]
+        }
+    },
+    "collection": "ml-model"
+}

main/model.safetensor

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b5ee01869aaaa14d65f86a81d9a2a825cc62663f2e6c9d79d5e687c7d49518d6
+size 188160632

main/sr_hard_constraint.safetensor

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fbad981519066387c413ead1d6af7ef3e0d2947c34147ba90163fc79ae539239
+size 1048656

main/sr_model.safetensor

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7da759a8479c6a821007b836a5dcb8ac988d18d3e1162f28a09a7c4feb47e7fa
+size 1130479968