Upload mlm.yaml

mlm.yaml

@@ -0,0 +1,38 @@
+$schema: "https://stac-extensions.github.io/mlm/v1.5.0/schema.json"
+properties:
+  name: FTW 3 Class Unet
+  architecture: torchgeo.models.unet
+  artifact_type: torch.export.save
+  framework: torch
+  framework_version: 2.8.0
+  accelerator: cuda
+  total_parameters: 1234567
+  tasks: [semantic-segmentation]
+  input:
+    - name: Imagery
+      bands: [B04, B03, B02, B08, B04, B03, B02, B08]
+      input:
+        shape: [-1, 8, -1, -1]
+        dim_order: [batch, channel, height, width]
+        data_type: float32
+      pre_processing_function:
+        format: torch.export.load
+        expression:
+          id: export-transform
+          name: transforms
+          href: path/to/archive.pt2
+          type: "application/octet-stream; framework=pytorch; profile=ExportedProgram"
+  output:
+    - name: segmentation-output
+      tasks: [semantic-segmentation]
+      result:
+        shape: [-1, 3, -1, -1]
+        dim_order: [batch, classes, height, width]
+        data_type: float32
+      classification:classes:
+        - value: 0
+          name: background
+        - value: 1
+          name: field
+        - value: 2
+          name: field-boundary