{
   "describedBy": "https://raw.githubusercontent.com/AllenNeuralDynamics/aind-data-schema/main/src/aind_data_schema/core/model.py",
   "schema_version": "0.0.1",
   "name": "model.safetensors",
   "license": "CC-BY-4.0",
   "developer_full_name": [
      "Camilo Laiton"
   ],
   "developer_institution": {
      "name": "Allen Institute for Neural Dynamics",
      "abbreviation": "AIND",
      "registry": {
         "name": "Research Organization Registry",
         "abbreviation": "ROR"
      },
      "registry_identifier": "04szwah67"
   },
   "modality": [
      {
         "name": "Selective plane illumination microscopy",
         "abbreviation": "SPIM"
      }
   ],
   "architecture": {
      "backbone": "UNet",
      "software": [
         {
            "name": "lightning",
            "version": "2.5.0.post0",
            "url": null,
            "parameters": {}
         }
      ],
      "layers": null,
      "parameters": {
         "image_width": 1024,
         "image_height": 1024
      },
      "notes": null
   },
   "intended_use": "2D tissue segmentation from SmartSPIM datasets",
   "limitations": "Only trained on 639 channel",
   "pretrained_source_url": null,
   "training": [
      {
         "name": "Model training",
         "software_version": null,
         "start_date_time": "2025-03-01T12:00:00-08:00",
         "end_date_time": "2025-03-01T14:00:00-08:00",
         "input_location": [
            "s3://aind-msma-morphology-data/test_data/SmartSPIM/brain_segmentation_2d_allaxis_channel_639"
         ],
         "output_location": "Path to deployment",
         "code_url": "https://github.com/AllenNeuralDynamics/aind-brain-segmentation",
         "code_version": "77d5159",
         "parameters": {
            "learning_rate": 0.00001,
            "batch_size": 32,
            "augmentation": true
         },
         "outputs": {},
         "notes": "Slices were pulled from a multiscaled image (14.4 microns in XY and 16.0 microns in Z) and then resized to 1024x1024",
         "resources": null,
         "train_performance": [
            {
               "name": "dice",
               "value": 0.96
            },
            {
               "name": "jaccard",
               "value": 0.93
            }
         ],
         "test_performance": [
            {
               "name": "dice",
               "value": 0.95
            },
            {
               "name": "jaccard",
               "value": 0.91
            }
         ],
         "test_data": "4:1 train/test split"
      }
   ],
   "evaluations": [],
   "notes": "model version 77d5159"
}