Hugging Face's logo

cellmap
/
onnx-web-models

ONNX
electron-microscopy
segmentation
cellmap
webgpu
onnxruntime-web
Model card Files
xet
 Community
onnx-web-models / setup_55 /finetune /torch_params.json
zouinkhim's picture
zouinkhim
3D full finetuning: NDHWC train_graph.json + torch_params.json, structured weight_map to_onnx, 3D runtime; 2D spike peak_mb corrected
3331842 verified
Raw History Blame Contribute Delete
14.3 kB
{
"encoder0.conv1.weight": {
"shape": [
32,
1,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"encoder0.conv1.convs.0.weight",
"encoder0.conv1.convs.1.weight",
"encoder0.conv1.convs.2.weight"
],
"axis": 2
}
]
},
"encoder0.conv1.bias": {
"shape": [
32
],
"from_onnx": [
{
"from": "encoder0.conv1.convs.0.bias"
}
]
},
"encoder0.bn1.weight": {
"shape": [
32
],
"from_onnx": [
{
"from": "encoder0.bn1.weight"
}
]
},
"encoder0.bn1.bias": {
"shape": [
32
],
"from_onnx": [
{
"from": "encoder0.bn1.bias"
}
]
},
"encoder0.bn1.running_mean": {
"shape": [
32
],
"from_onnx": [
{
"from": "encoder0.bn1.running_mean"
}
]
},
"encoder0.bn1.running_var": {
"shape": [
32
],
"from_onnx": [
{
"from": "encoder0.bn1.running_var"
}
]
},
"encoder0.conv2.weight": {
"shape": [
32,
32,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"encoder0.conv2.convs.0.weight",
"encoder0.conv2.convs.1.weight",
"encoder0.conv2.convs.2.weight"
],
"axis": 2
}
]
},
"encoder0.conv2.bias": {
"shape": [
32
],
"from_onnx": [
{
"from": "encoder0.conv2.convs.0.bias"
}
]
},
"encoder0.bn2.weight": {
"shape": [
32
],
"from_onnx": [
{
"from": "encoder0.bn2.weight"
}
]
},
"encoder0.bn2.bias": {
"shape": [
32
],
"from_onnx": [
{
"from": "encoder0.bn2.bias"
}
]
},
"encoder0.bn2.running_mean": {
"shape": [
32
],
"from_onnx": [
{
"from": "encoder0.bn2.running_mean"
}
]
},
"encoder0.bn2.running_var": {
"shape": [
32
],
"from_onnx": [
{
"from": "encoder0.bn2.running_var"
}
]
},
"encoder1.conv1.weight": {
"shape": [
64,
32,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"encoder1.conv1.convs.0.weight",
"encoder1.conv1.convs.1.weight",
"encoder1.conv1.convs.2.weight"
],
"axis": 2
}
]
},
"encoder1.conv1.bias": {
"shape": [
64
],
"from_onnx": [
{
"from": "encoder1.conv1.convs.0.bias"
}
]
},
"encoder1.bn1.weight": {
"shape": [
64
],
"from_onnx": [
{
"from": "encoder1.bn1.weight"
}
]
},
"encoder1.bn1.bias": {
"shape": [
64
],
"from_onnx": [
{
"from": "encoder1.bn1.bias"
}
]
},
"encoder1.bn1.running_mean": {
"shape": [
64
],
"from_onnx": [
{
"from": "encoder1.bn1.running_mean"
}
]
},
"encoder1.bn1.running_var": {
"shape": [
64
],
"from_onnx": [
{
"from": "encoder1.bn1.running_var"
}
]
},
"encoder1.conv2.weight": {
"shape": [
64,
64,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"encoder1.conv2.convs.0.weight",
"encoder1.conv2.convs.1.weight",
"encoder1.conv2.convs.2.weight"
],
"axis": 2
}
]
},
"encoder1.conv2.bias": {
"shape": [
64
],
"from_onnx": [
{
"from": "encoder1.conv2.convs.0.bias"
}
]
},
"encoder1.bn2.weight": {
"shape": [
64
],
"from_onnx": [
{
"from": "encoder1.bn2.weight"
}
]
},
"encoder1.bn2.bias": {
"shape": [
64
],
"from_onnx": [
{
"from": "encoder1.bn2.bias"
}
]
},
"encoder1.bn2.running_mean": {
"shape": [
64
],
"from_onnx": [
{
"from": "encoder1.bn2.running_mean"
}
]
},
"encoder1.bn2.running_var": {
"shape": [
64
],
"from_onnx": [
{
"from": "encoder1.bn2.running_var"
}
]
},
"encoder2.conv1.weight": {
"shape": [
128,
64,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"encoder2.conv1.convs.0.weight",
"encoder2.conv1.convs.1.weight",
"encoder2.conv1.convs.2.weight"
],
"axis": 2
}
]
},
"encoder2.conv1.bias": {
"shape": [
128
],
"from_onnx": [
{
"from": "encoder2.conv1.convs.0.bias"
}
]
},
"encoder2.bn1.weight": {
"shape": [
128
],
"from_onnx": [
{
"from": "encoder2.bn1.weight"
}
]
},
"encoder2.bn1.bias": {
"shape": [
128
],
"from_onnx": [
{
"from": "encoder2.bn1.bias"
}
]
},
"encoder2.bn1.running_mean": {
"shape": [
128
],
"from_onnx": [
{
"from": "encoder2.bn1.running_mean"
}
]
},
"encoder2.bn1.running_var": {
"shape": [
128
],
"from_onnx": [
{
"from": "encoder2.bn1.running_var"
}
]
},
"encoder2.conv2.weight": {
"shape": [
128,
128,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"encoder2.conv2.convs.0.weight",
"encoder2.conv2.convs.1.weight",
"encoder2.conv2.convs.2.weight"
],
"axis": 2
}
]
},
"encoder2.conv2.bias": {
"shape": [
128
],
"from_onnx": [
{
"from": "encoder2.conv2.convs.0.bias"
}
]
},
"encoder2.bn2.weight": {
"shape": [
128
],
"from_onnx": [
{
"from": "encoder2.bn2.weight"
}
]
},
"encoder2.bn2.bias": {
"shape": [
128
],
"from_onnx": [
{
"from": "encoder2.bn2.bias"
}
]
},
"encoder2.bn2.running_mean": {
"shape": [
128
],
"from_onnx": [
{
"from": "encoder2.bn2.running_mean"
}
]
},
"encoder2.bn2.running_var": {
"shape": [
128
],
"from_onnx": [
{
"from": "encoder2.bn2.running_var"
}
]
},
"bottleneck.conv1.weight": {
"shape": [
256,
128,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"bottleneck.conv1.convs.0.weight",
"bottleneck.conv1.convs.1.weight",
"bottleneck.conv1.convs.2.weight"
],
"axis": 2
}
]
},
"bottleneck.conv1.bias": {
"shape": [
256
],
"from_onnx": [
{
"from": "bottleneck.conv1.convs.0.bias"
}
]
},
"bottleneck.bn1.weight": {
"shape": [
256
],
"from_onnx": [
{
"from": "bottleneck.bn1.weight"
}
]
},
"bottleneck.bn1.bias": {
"shape": [
256
],
"from_onnx": [
{
"from": "bottleneck.bn1.bias"
}
]
},
"bottleneck.bn1.running_mean": {
"shape": [
256
],
"from_onnx": [
{
"from": "bottleneck.bn1.running_mean"
}
]
},
"bottleneck.bn1.running_var": {
"shape": [
256
],
"from_onnx": [
{
"from": "bottleneck.bn1.running_var"
}
]
},
"bottleneck.conv2.weight": {
"shape": [
256,
256,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"bottleneck.conv2.convs.0.weight",
"bottleneck.conv2.convs.1.weight",
"bottleneck.conv2.convs.2.weight"
],
"axis": 2
}
]
},
"bottleneck.conv2.bias": {
"shape": [
256
],
"from_onnx": [
{
"from": "bottleneck.conv2.convs.0.bias"
}
]
},
"bottleneck.bn2.weight": {
"shape": [
256
],
"from_onnx": [
{
"from": "bottleneck.bn2.weight"
}
]
},
"bottleneck.bn2.bias": {
"shape": [
256
],
"from_onnx": [
{
"from": "bottleneck.bn2.bias"
}
]
},
"bottleneck.bn2.running_mean": {
"shape": [
256
],
"from_onnx": [
{
"from": "bottleneck.bn2.running_mean"
}
]
},
"bottleneck.bn2.running_var": {
"shape": [
256
],
"from_onnx": [
{
"from": "bottleneck.bn2.running_var"
}
]
},
"upconv2.weight": {
"shape": [
256,
128,
2,
2,
2
],
"from_onnx": [
{
"from": "upconv2.conv.convs.0.weight"
},
{
"reshape": [
128,
2,
2,
2,
256
]
},
{
"permute": [
4,
0,
1,
2,
3
]
}
]
},
"upconv2.bias": {
"shape": [
128
],
"from_onnx": [
{
"from": "upconv2.conv.convs.0.bias"
},
{
"reshape": [
128,
8
]
},
{
"index": [
1,
0
]
}
]
},
"decoder2.conv1.weight": {
"shape": [
128,
256,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"decoder2.conv1.convs.0.weight",
"decoder2.conv1.convs.1.weight",
"decoder2.conv1.convs.2.weight"
],
"axis": 2
}
]
},
"decoder2.conv1.bias": {
"shape": [
128
],
"from_onnx": [
{
"from": "decoder2.conv1.convs.0.bias"
}
]
},
"decoder2.bn1.weight": {
"shape": [
128
],
"from_onnx": [
{
"from": "decoder2.bn1.weight"
}
]
},
"decoder2.bn1.bias": {
"shape": [
128
],
"from_onnx": [
{
"from": "decoder2.bn1.bias"
}
]
},
"decoder2.bn1.running_mean": {
"shape": [
128
],
"from_onnx": [
{
"from": "decoder2.bn1.running_mean"
}
]
},
"decoder2.bn1.running_var": {
"shape": [
128
],
"from_onnx": [
{
"from": "decoder2.bn1.running_var"
}
]
},
"decoder2.conv2.weight": {
"shape": [
128,
128,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"decoder2.conv2.convs.0.weight",
"decoder2.conv2.convs.1.weight",
"decoder2.conv2.convs.2.weight"
],
"axis": 2
}
]
},
"decoder2.conv2.bias": {
"shape": [
128
],
"from_onnx": [
{
"from": "decoder2.conv2.convs.0.bias"
}
]
},
"decoder2.bn2.weight": {
"shape": [
128
],
"from_onnx": [
{
"from": "decoder2.bn2.weight"
}
]
},
"decoder2.bn2.bias": {
"shape": [
128
],
"from_onnx": [
{
"from": "decoder2.bn2.bias"
}
]
},
"decoder2.bn2.running_mean": {
"shape": [
128
],
"from_onnx": [
{
"from": "decoder2.bn2.running_mean"
}
]
},
"decoder2.bn2.running_var": {
"shape": [
128
],
"from_onnx": [
{
"from": "decoder2.bn2.running_var"
}
]
},
"upconv1.weight": {
"shape": [
128,
64,
2,
2,
2
],
"from_onnx": [
{
"from": "upconv1.conv.convs.0.weight"
},
{
"reshape": [
64,
2,
2,
2,
128
]
},
{
"permute": [
4,
0,
1,
2,
3
]
}
]
},
"upconv1.bias": {
"shape": [
64
],
"from_onnx": [
{
"from": "upconv1.conv.convs.0.bias"
},
{
"reshape": [
64,
8
]
},
{
"index": [
1,
0
]
}
]
},
"decoder1.conv1.weight": {
"shape": [
64,
128,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"decoder1.conv1.convs.0.weight",
"decoder1.conv1.convs.1.weight",
"decoder1.conv1.convs.2.weight"
],
"axis": 2
}
]
},
"decoder1.conv1.bias": {
"shape": [
64
],
"from_onnx": [
{
"from": "decoder1.conv1.convs.0.bias"
}
]
},
"decoder1.bn1.weight": {
"shape": [
64
],
"from_onnx": [
{
"from": "decoder1.bn1.weight"
}
]
},
"decoder1.bn1.bias": {
"shape": [
64
],
"from_onnx": [
{
"from": "decoder1.bn1.bias"
}
]
},
"decoder1.bn1.running_mean": {
"shape": [
64
],
"from_onnx": [
{
"from": "decoder1.bn1.running_mean"
}
]
},
"decoder1.bn1.running_var": {
"shape": [
64
],
"from_onnx": [
{
"from": "decoder1.bn1.running_var"
}
]
},
"decoder1.conv2.weight": {
"shape": [
64,
64,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"decoder1.conv2.convs.0.weight",
"decoder1.conv2.convs.1.weight",
"decoder1.conv2.convs.2.weight"
],
"axis": 2
}
]
},
"decoder1.conv2.bias": {
"shape": [
64
],
"from_onnx": [
{
"from": "decoder1.conv2.convs.0.bias"
}
]
},
"decoder1.bn2.weight": {
"shape": [
64
],
"from_onnx": [
{
"from": "decoder1.bn2.weight"
}
]
},
"decoder1.bn2.bias": {
"shape": [
64
],
"from_onnx": [
{
"from": "decoder1.bn2.bias"
}
]
},
"decoder1.bn2.running_mean": {
"shape": [
64
],
"from_onnx": [
{
"from": "decoder1.bn2.running_mean"
}
]
},
"decoder1.bn2.running_var": {
"shape": [
64
],
"from_onnx": [
{
"from": "decoder1.bn2.running_var"
}
]
},
"upconv0.weight": {
"shape": [
64,
32,
2,
2,
2
],
"from_onnx": [
{
"from": "upconv0.conv.convs.0.weight"
},
{
"reshape": [
32,
2,
2,
2,
64
]
},
{
"permute": [
4,
0,
1,
2,
3
]
}
]
},
"upconv0.bias": {
"shape": [
32
],
"from_onnx": [
{
"from": "upconv0.conv.convs.0.bias"
},
{
"reshape": [
32,
8
]
},
{
"index": [
1,
0
]
}
]
},
"decoder0.conv1.weight": {
"shape": [
32,
64,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"decoder0.conv1.convs.0.weight",
"decoder0.conv1.convs.1.weight",
"decoder0.conv1.convs.2.weight"
],
"axis": 2
}
]
},
"decoder0.conv1.bias": {
"shape": [
32
],
"from_onnx": [
{
"from": "decoder0.conv1.convs.0.bias"
}
]
},
"decoder0.bn1.weight": {
"shape": [
32
],
"from_onnx": [
{
"from": "decoder0.bn1.weight"
}
]
},
"decoder0.bn1.bias": {
"shape": [
32
],
"from_onnx": [
{
"from": "decoder0.bn1.bias"
}
]
},
"decoder0.bn1.running_mean": {
"shape": [
32
],
"from_onnx": [
{
"from": "decoder0.bn1.running_mean"
}
]
},
"decoder0.bn1.running_var": {
"shape": [
32
],
"from_onnx": [
{
"from": "decoder0.bn1.running_var"
}
]
},
"decoder0.conv2.weight": {
"shape": [
32,
32,
3,
3,
3
],
"from_onnx": [
{
"stack": [
"decoder0.conv2.convs.0.weight",
"decoder0.conv2.convs.1.weight",
"decoder0.conv2.convs.2.weight"
],
"axis": 2
}
]
},
"decoder0.conv2.bias": {
"shape": [
32
],
"from_onnx": [
{
"from": "decoder0.conv2.convs.0.bias"
}
]
},
"decoder0.bn2.weight": {
"shape": [
32
],
"from_onnx": [
{
"from": "decoder0.bn2.weight"
}
]
},
"decoder0.bn2.bias": {
"shape": [
32
],
"from_onnx": [
{
"from": "decoder0.bn2.bias"
}
]
},
"decoder0.bn2.running_mean": {
"shape": [
32
],
"from_onnx": [
{
"from": "decoder0.bn2.running_mean"
}
]
},
"decoder0.bn2.running_var": {
"shape": [
32
],
"from_onnx": [
{
"from": "decoder0.bn2.running_var"
}
]
},
"final_conv.weight": {
"shape": [
1,
32,
1,
1,
1
],
"from_onnx": [
{
"stack": [
"final_conv.convs.0.weight"
],
"axis": 2
}
]
},
"final_conv.bias": {
"shape": [
1
],
"from_onnx": [
{
"from": "final_conv.convs.0.bias"
}
]
}
}