Hugging Face's logo

xiao-1011
/
dmc-brainmap-registration-predictor

Keypoint Detection
PyTorch
computer-vision
image-registration
homography-estimation
paired-image
neuroscience
histology
brain-atlas
napari
Model card Files
xet
 Community
dmc-brainmap-registration-predictor
File size: 1,930 Bytes
c2021af
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2e8a8aa
 
 
 
 
 
c2021af
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
{
  "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"
  }
}