{
    "type": "Feature",
    "stac_version": "1.1.0",
    "stac_extensions": [
        "https://stac-extensions.github.io/mlm/v1.4.0/schema.json"
    ],
    "id": "SEN2SRLite_NonReference_RGBN_x4",
    "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 Swift Parameter-free Attention Network (SPAN) trained on the SEN2NAIPv2 dataset to enhance RGBN Sentinel-2 bands, improving spatial resolution from 10 meters to 2.5 meters.",
        "dependencies": [
            "torch",
            "safetensors.torch",
            "sen2sr"
        ],
        "mlm:framework": "pytorch",
        "mlm:framework_version": "2.1.2+cu121",
        "file:size": 1889984,
        "mlm:memory_size": 1,
        "mlm:accelerator": "cuda",
        "mlm:accelerator_constrained": false,
        "mlm:accelerator_summary": "Unknown",
        "mlm:name": "CNN_Light_SR",
        "mlm:architecture": "SPAN",
        "mlm:tasks": [
            "super-resolution"
        ],
        "mlm:input": [
            {
                "name": "4 Band Sentinel-2 10m bands",
                "bands": [
                    "B04",
                    "B03",
                    "B02",
                    "B08"
                ],
                "input": {
                    "shape": [
                        -1,
                        4,
                        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,
                        4,
                        512,
                        512
                    ],
                    "dim_order": [
                        "batch",
                        "channel",
                        "height",
                        "width"
                    ],
                    "data_type": "float16"
                },
                "classification:classes": [],
                "post_processing_function": null
            }
        ],
        "mlm:total_parameters": 472496,
        "mlm:pretrained": true,
        "datetime": null
    },
    "links": [],
    "assets": {
        "trainable": {
            "href": "https://huggingface.co/tacofoundation/SEN2SR/resolve/main/SEN2SRLite/NonReference_RGBN_x4/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/SEN2SRLite/NonReference_RGBN_x4/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/SEN2SRLite/NonReference_RGBN_x4/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/SEN2SRLite/NonReference_RGBN_x4/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"
            ]
        }
    },
    "collection": "SEN2SRLite"
}