| { |
| "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" |
| } |
