| { |
| "format_version": 1, |
| "model_id": "xiao-1011/dmc-brainmap-registration-predictor", |
| "model_version": "1.0.0", |
| "architecture": { |
| "class_name": "PairSpatialHomographyNet", |
| "parameter_count": 2336200, |
| "input_channels": 2, |
| "base_channels": 24, |
| "feature_channels": [24, 48, 96, 96, 192], |
| "spatial_pool_size": [5, 8], |
| "spatial_hidden_channels": 256, |
| "max_corner_offset_px": 512.0 |
| }, |
| "inputs": [ |
| { |
| "name": "sample", |
| "shape": ["batch", 1, 800, 1140], |
| "dtype": "float32", |
| "range": [0.0, 1.0] |
| }, |
| { |
| "name": "reference", |
| "shape": ["batch", 1, 800, 1140], |
| "dtype": "float32", |
| "range": [0.0, 1.0] |
| } |
| ], |
| "preprocessing": { |
| "color_mode": "grayscale", |
| "pixel_scale_divisor": 255.0, |
| "tensor_layout": "NCHW", |
| "validated_width_px": 1140, |
| "validated_height_px": 800 |
| }, |
| "output": { |
| "name": "sample_to_reference_corner_offsets", |
| "shape": ["batch", 4, 2], |
| "dtype": "float32", |
| "unit": "pixel", |
| "direction": "sample_to_reference", |
| "corner_order": ["top_left", "top_right", "bottom_right", "bottom_left"] |
| }, |
| "compatibility": { |
| "framework": "pytorch", |
| "training_torch_version": "2.12.0+cu130", |
| "inference_device": "cpu", |
| "application": "napari-dmc-brainmap" |
| }, |
| "artifact": { |
| "filename": "dmc-brainmap-registration-predictor-v1.0.0.pt", |
| "size_bytes": 9363033, |
| "sha256": "f24893d0a5d335ccfee2c354245cfefef935424f0d7c8010e9efbc0560496d18", |
| "weights_only_compatible": true |
| }, |
| "provenance": { |
| "source_repository": "https://github.com/xiao-1011/sharpy-model", |
| "source_commit": "0a80f4f831f6450a015a2f32433f22975ac97dc0", |
| "run_name": "baseline-005-spatial-augmented-200ep", |
| "selected_epoch": 91, |
| "selection_metric": "validation_loss", |
| "original_checkpoint_sha256": "2f6c84b12a8449a63ff45973cb91a81326973915d4801659c99b973087c1dac8" |
| } |
| } |
|
|
