Download setup_55/finetune/train_graph.json from cellmap/onnx-web-models: direct link, hf CLI and curl.
- Browser
- Download file 20 kB
-
https://huggingface.co/cellmap/onnx-web-models/resolve/main/setup_55/finetune/train_graph.json
- Command line
-
hf download hf://cellmap/onnx-web-models/setup_55/finetune/train_graph.json
-
curl -L -o train_graph.json https://huggingface.co/cellmap/onnx-web-models/resolve/main/setup_55/finetune/train_graph.json
20 kB
| { | |
| "format": "cellmap-train-graph/1", | |
| "layout": "NDHWC", | |
| "input": "input", | |
| "output": "output", | |
| "context": 88, | |
| "params": { | |
| "encoder0.conv1.weight": { | |
| "shape": [ | |
| 32, | |
| 1, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder0.conv1.bias": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder0.bn1.weight": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder0.bn1.bias": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder0.bn1.running_mean": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": false | |
| }, | |
| "encoder0.bn1.running_var": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": false | |
| }, | |
| "encoder0.conv2.weight": { | |
| "shape": [ | |
| 32, | |
| 32, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder0.conv2.bias": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder0.bn2.weight": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder0.bn2.bias": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder0.bn2.running_mean": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": false | |
| }, | |
| "encoder0.bn2.running_var": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": false | |
| }, | |
| "encoder1.conv1.weight": { | |
| "shape": [ | |
| 64, | |
| 32, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder1.conv1.bias": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder1.bn1.weight": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder1.bn1.bias": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder1.bn1.running_mean": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": false | |
| }, | |
| "encoder1.bn1.running_var": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": false | |
| }, | |
| "encoder1.conv2.weight": { | |
| "shape": [ | |
| 64, | |
| 64, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder1.conv2.bias": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder1.bn2.weight": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder1.bn2.bias": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder1.bn2.running_mean": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": false | |
| }, | |
| "encoder1.bn2.running_var": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": false | |
| }, | |
| "encoder2.conv1.weight": { | |
| "shape": [ | |
| 128, | |
| 64, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder2.conv1.bias": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder2.bn1.weight": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder2.bn1.bias": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder2.bn1.running_mean": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": false | |
| }, | |
| "encoder2.bn1.running_var": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": false | |
| }, | |
| "encoder2.conv2.weight": { | |
| "shape": [ | |
| 128, | |
| 128, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder2.conv2.bias": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder2.bn2.weight": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder2.bn2.bias": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "encoder2.bn2.running_mean": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": false | |
| }, | |
| "encoder2.bn2.running_var": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": false | |
| }, | |
| "bottleneck.conv1.weight": { | |
| "shape": [ | |
| 256, | |
| 128, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "bottleneck.conv1.bias": { | |
| "shape": [ | |
| 256 | |
| ], | |
| "trainable": true | |
| }, | |
| "bottleneck.bn1.weight": { | |
| "shape": [ | |
| 256 | |
| ], | |
| "trainable": true | |
| }, | |
| "bottleneck.bn1.bias": { | |
| "shape": [ | |
| 256 | |
| ], | |
| "trainable": true | |
| }, | |
| "bottleneck.bn1.running_mean": { | |
| "shape": [ | |
| 256 | |
| ], | |
| "trainable": false | |
| }, | |
| "bottleneck.bn1.running_var": { | |
| "shape": [ | |
| 256 | |
| ], | |
| "trainable": false | |
| }, | |
| "bottleneck.conv2.weight": { | |
| "shape": [ | |
| 256, | |
| 256, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "bottleneck.conv2.bias": { | |
| "shape": [ | |
| 256 | |
| ], | |
| "trainable": true | |
| }, | |
| "bottleneck.bn2.weight": { | |
| "shape": [ | |
| 256 | |
| ], | |
| "trainable": true | |
| }, | |
| "bottleneck.bn2.bias": { | |
| "shape": [ | |
| 256 | |
| ], | |
| "trainable": true | |
| }, | |
| "bottleneck.bn2.running_mean": { | |
| "shape": [ | |
| 256 | |
| ], | |
| "trainable": false | |
| }, | |
| "bottleneck.bn2.running_var": { | |
| "shape": [ | |
| 256 | |
| ], | |
| "trainable": false | |
| }, | |
| "upconv2.weight": { | |
| "shape": [ | |
| 256, | |
| 128, | |
| 2, | |
| 2, | |
| 2 | |
| ], | |
| "trainable": true | |
| }, | |
| "upconv2.bias": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder2.conv1.weight": { | |
| "shape": [ | |
| 128, | |
| 256, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder2.conv1.bias": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder2.bn1.weight": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder2.bn1.bias": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder2.bn1.running_mean": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": false | |
| }, | |
| "decoder2.bn1.running_var": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": false | |
| }, | |
| "decoder2.conv2.weight": { | |
| "shape": [ | |
| 128, | |
| 128, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder2.conv2.bias": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder2.bn2.weight": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder2.bn2.bias": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder2.bn2.running_mean": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": false | |
| }, | |
| "decoder2.bn2.running_var": { | |
| "shape": [ | |
| 128 | |
| ], | |
| "trainable": false | |
| }, | |
| "upconv1.weight": { | |
| "shape": [ | |
| 128, | |
| 64, | |
| 2, | |
| 2, | |
| 2 | |
| ], | |
| "trainable": true | |
| }, | |
| "upconv1.bias": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder1.conv1.weight": { | |
| "shape": [ | |
| 64, | |
| 128, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder1.conv1.bias": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder1.bn1.weight": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder1.bn1.bias": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder1.bn1.running_mean": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": false | |
| }, | |
| "decoder1.bn1.running_var": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": false | |
| }, | |
| "decoder1.conv2.weight": { | |
| "shape": [ | |
| 64, | |
| 64, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder1.conv2.bias": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder1.bn2.weight": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder1.bn2.bias": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder1.bn2.running_mean": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": false | |
| }, | |
| "decoder1.bn2.running_var": { | |
| "shape": [ | |
| 64 | |
| ], | |
| "trainable": false | |
| }, | |
| "upconv0.weight": { | |
| "shape": [ | |
| 64, | |
| 32, | |
| 2, | |
| 2, | |
| 2 | |
| ], | |
| "trainable": true | |
| }, | |
| "upconv0.bias": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder0.conv1.weight": { | |
| "shape": [ | |
| 32, | |
| 64, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder0.conv1.bias": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder0.bn1.weight": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder0.bn1.bias": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder0.bn1.running_mean": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": false | |
| }, | |
| "decoder0.bn1.running_var": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": false | |
| }, | |
| "decoder0.conv2.weight": { | |
| "shape": [ | |
| 32, | |
| 32, | |
| 3, | |
| 3, | |
| 3 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder0.conv2.bias": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder0.bn2.weight": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder0.bn2.bias": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": true | |
| }, | |
| "decoder0.bn2.running_mean": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": false | |
| }, | |
| "decoder0.bn2.running_var": { | |
| "shape": [ | |
| 32 | |
| ], | |
| "trainable": false | |
| }, | |
| "final_conv.weight": { | |
| "shape": [ | |
| 1, | |
| 32, | |
| 1, | |
| 1, | |
| 1 | |
| ], | |
| "trainable": true | |
| }, | |
| "final_conv.bias": { | |
| "shape": [ | |
| 1 | |
| ], | |
| "trainable": true | |
| } | |
| }, | |
| "ops": [ | |
| { | |
| "op": "conv", | |
| "x": "input", | |
| "y": "/encoder0/conv1/Conv_output_0", | |
| "w": "encoder0.conv1.weight", | |
| "b": "encoder0.conv1.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/encoder0/conv1/Conv_output_0", | |
| "y": "/encoder0/bn1/BatchNormalization_output_0", | |
| "scale": "encoder0.bn1.weight", | |
| "bias": "encoder0.bn1.bias", | |
| "mean": "encoder0.bn1.running_mean", | |
| "var": "encoder0.bn1.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/encoder0/bn1/BatchNormalization_output_0", | |
| "y": "/encoder0/act/Relu_output_0" | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/encoder0/act/Relu_output_0", | |
| "y": "/encoder0/conv2/Conv_output_0", | |
| "w": "encoder0.conv2.weight", | |
| "b": "encoder0.conv2.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/encoder0/conv2/Conv_output_0", | |
| "y": "/encoder0/bn2/BatchNormalization_output_0", | |
| "scale": "encoder0.bn2.weight", | |
| "bias": "encoder0.bn2.bias", | |
| "mean": "encoder0.bn2.running_mean", | |
| "var": "encoder0.bn2.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/encoder0/bn2/BatchNormalization_output_0", | |
| "y": "/encoder0/act_1/Relu_output_0" | |
| }, | |
| { | |
| "op": "maxpool", | |
| "x": "/encoder0/act_1/Relu_output_0", | |
| "y": "/pool1/MaxPool_output_0", | |
| "k": [ | |
| 2, | |
| 2, | |
| 2 | |
| ], | |
| "stride": [ | |
| 2, | |
| 2, | |
| 2 | |
| ] | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/pool1/MaxPool_output_0", | |
| "y": "/encoder1/conv1/Conv_output_0", | |
| "w": "encoder1.conv1.weight", | |
| "b": "encoder1.conv1.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/encoder1/conv1/Conv_output_0", | |
| "y": "/encoder1/bn1/BatchNormalization_output_0", | |
| "scale": "encoder1.bn1.weight", | |
| "bias": "encoder1.bn1.bias", | |
| "mean": "encoder1.bn1.running_mean", | |
| "var": "encoder1.bn1.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/encoder1/bn1/BatchNormalization_output_0", | |
| "y": "/encoder1/act/Relu_output_0" | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/encoder1/act/Relu_output_0", | |
| "y": "/encoder1/conv2/Conv_output_0", | |
| "w": "encoder1.conv2.weight", | |
| "b": "encoder1.conv2.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/encoder1/conv2/Conv_output_0", | |
| "y": "/encoder1/bn2/BatchNormalization_output_0", | |
| "scale": "encoder1.bn2.weight", | |
| "bias": "encoder1.bn2.bias", | |
| "mean": "encoder1.bn2.running_mean", | |
| "var": "encoder1.bn2.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/encoder1/bn2/BatchNormalization_output_0", | |
| "y": "/encoder1/act_1/Relu_output_0" | |
| }, | |
| { | |
| "op": "maxpool", | |
| "x": "/encoder1/act_1/Relu_output_0", | |
| "y": "/pool2/MaxPool_output_0", | |
| "k": [ | |
| 2, | |
| 2, | |
| 2 | |
| ], | |
| "stride": [ | |
| 2, | |
| 2, | |
| 2 | |
| ] | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/pool2/MaxPool_output_0", | |
| "y": "/encoder2/conv1/Conv_output_0", | |
| "w": "encoder2.conv1.weight", | |
| "b": "encoder2.conv1.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/encoder2/conv1/Conv_output_0", | |
| "y": "/encoder2/bn1/BatchNormalization_output_0", | |
| "scale": "encoder2.bn1.weight", | |
| "bias": "encoder2.bn1.bias", | |
| "mean": "encoder2.bn1.running_mean", | |
| "var": "encoder2.bn1.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/encoder2/bn1/BatchNormalization_output_0", | |
| "y": "/encoder2/act/Relu_output_0" | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/encoder2/act/Relu_output_0", | |
| "y": "/encoder2/conv2/Conv_output_0", | |
| "w": "encoder2.conv2.weight", | |
| "b": "encoder2.conv2.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/encoder2/conv2/Conv_output_0", | |
| "y": "/encoder2/bn2/BatchNormalization_output_0", | |
| "scale": "encoder2.bn2.weight", | |
| "bias": "encoder2.bn2.bias", | |
| "mean": "encoder2.bn2.running_mean", | |
| "var": "encoder2.bn2.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/encoder2/bn2/BatchNormalization_output_0", | |
| "y": "/encoder2/act_1/Relu_output_0" | |
| }, | |
| { | |
| "op": "maxpool", | |
| "x": "/encoder2/act_1/Relu_output_0", | |
| "y": "/pool3/MaxPool_output_0", | |
| "k": [ | |
| 2, | |
| 2, | |
| 2 | |
| ], | |
| "stride": [ | |
| 2, | |
| 2, | |
| 2 | |
| ] | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/pool3/MaxPool_output_0", | |
| "y": "/bottleneck/conv1/Conv_output_0", | |
| "w": "bottleneck.conv1.weight", | |
| "b": "bottleneck.conv1.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/bottleneck/conv1/Conv_output_0", | |
| "y": "/bottleneck/bn1/BatchNormalization_output_0", | |
| "scale": "bottleneck.bn1.weight", | |
| "bias": "bottleneck.bn1.bias", | |
| "mean": "bottleneck.bn1.running_mean", | |
| "var": "bottleneck.bn1.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/bottleneck/bn1/BatchNormalization_output_0", | |
| "y": "/bottleneck/act/Relu_output_0" | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/bottleneck/act/Relu_output_0", | |
| "y": "/bottleneck/conv2/Conv_output_0", | |
| "w": "bottleneck.conv2.weight", | |
| "b": "bottleneck.conv2.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/bottleneck/conv2/Conv_output_0", | |
| "y": "/bottleneck/bn2/BatchNormalization_output_0", | |
| "scale": "bottleneck.bn2.weight", | |
| "bias": "bottleneck.bn2.bias", | |
| "mean": "bottleneck.bn2.running_mean", | |
| "var": "bottleneck.bn2.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/bottleneck/bn2/BatchNormalization_output_0", | |
| "y": "/bottleneck/act_1/Relu_output_0" | |
| }, | |
| { | |
| "op": "conv_t", | |
| "x": "/bottleneck/act_1/Relu_output_0", | |
| "y": "/upconv2/ConvTranspose_output_0", | |
| "w": "upconv2.weight", | |
| "b": "upconv2.bias", | |
| "stride": [ | |
| 2, | |
| 2, | |
| 2 | |
| ] | |
| }, | |
| { | |
| "op": "crop", | |
| "x": "/encoder2/act_1/Relu_output_0", | |
| "y": "/cc2/Slice_output_0", | |
| "axis": "d", | |
| "start": 4, | |
| "end_margin": 4 | |
| }, | |
| { | |
| "op": "crop", | |
| "x": "/cc2/Slice_output_0", | |
| "y": "/cc2/Slice_1_output_0", | |
| "axis": "h", | |
| "start": 4, | |
| "end_margin": 4 | |
| }, | |
| { | |
| "op": "crop", | |
| "x": "/cc2/Slice_1_output_0", | |
| "y": "/cc2/Slice_2_output_0", | |
| "axis": "w", | |
| "start": 4, | |
| "end_margin": 4 | |
| }, | |
| { | |
| "op": "concat", | |
| "xs": [ | |
| "/cc2/Slice_2_output_0", | |
| "/upconv2/ConvTranspose_output_0" | |
| ], | |
| "y": "/cc2/Concat_output_0" | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/cc2/Concat_output_0", | |
| "y": "/decoder2/conv1/Conv_output_0", | |
| "w": "decoder2.conv1.weight", | |
| "b": "decoder2.conv1.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/decoder2/conv1/Conv_output_0", | |
| "y": "/decoder2/bn1/BatchNormalization_output_0", | |
| "scale": "decoder2.bn1.weight", | |
| "bias": "decoder2.bn1.bias", | |
| "mean": "decoder2.bn1.running_mean", | |
| "var": "decoder2.bn1.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/decoder2/bn1/BatchNormalization_output_0", | |
| "y": "/decoder2/act/Relu_output_0" | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/decoder2/act/Relu_output_0", | |
| "y": "/decoder2/conv2/Conv_output_0", | |
| "w": "decoder2.conv2.weight", | |
| "b": "decoder2.conv2.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/decoder2/conv2/Conv_output_0", | |
| "y": "/decoder2/bn2/BatchNormalization_output_0", | |
| "scale": "decoder2.bn2.weight", | |
| "bias": "decoder2.bn2.bias", | |
| "mean": "decoder2.bn2.running_mean", | |
| "var": "decoder2.bn2.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/decoder2/bn2/BatchNormalization_output_0", | |
| "y": "/decoder2/act_1/Relu_output_0" | |
| }, | |
| { | |
| "op": "conv_t", | |
| "x": "/decoder2/act_1/Relu_output_0", | |
| "y": "/upconv1/ConvTranspose_output_0", | |
| "w": "upconv1.weight", | |
| "b": "upconv1.bias", | |
| "stride": [ | |
| 2, | |
| 2, | |
| 2 | |
| ] | |
| }, | |
| { | |
| "op": "crop", | |
| "x": "/encoder1/act_1/Relu_output_0", | |
| "y": "/cc1/Slice_output_0", | |
| "axis": "d", | |
| "start": 16, | |
| "end_margin": 16 | |
| }, | |
| { | |
| "op": "crop", | |
| "x": "/cc1/Slice_output_0", | |
| "y": "/cc1/Slice_1_output_0", | |
| "axis": "h", | |
| "start": 16, | |
| "end_margin": 16 | |
| }, | |
| { | |
| "op": "crop", | |
| "x": "/cc1/Slice_1_output_0", | |
| "y": "/cc1/Slice_2_output_0", | |
| "axis": "w", | |
| "start": 16, | |
| "end_margin": 16 | |
| }, | |
| { | |
| "op": "concat", | |
| "xs": [ | |
| "/cc1/Slice_2_output_0", | |
| "/upconv1/ConvTranspose_output_0" | |
| ], | |
| "y": "/cc1/Concat_output_0" | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/cc1/Concat_output_0", | |
| "y": "/decoder1/conv1/Conv_output_0", | |
| "w": "decoder1.conv1.weight", | |
| "b": "decoder1.conv1.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/decoder1/conv1/Conv_output_0", | |
| "y": "/decoder1/bn1/BatchNormalization_output_0", | |
| "scale": "decoder1.bn1.weight", | |
| "bias": "decoder1.bn1.bias", | |
| "mean": "decoder1.bn1.running_mean", | |
| "var": "decoder1.bn1.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/decoder1/bn1/BatchNormalization_output_0", | |
| "y": "/decoder1/act/Relu_output_0" | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/decoder1/act/Relu_output_0", | |
| "y": "/decoder1/conv2/Conv_output_0", | |
| "w": "decoder1.conv2.weight", | |
| "b": "decoder1.conv2.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/decoder1/conv2/Conv_output_0", | |
| "y": "/decoder1/bn2/BatchNormalization_output_0", | |
| "scale": "decoder1.bn2.weight", | |
| "bias": "decoder1.bn2.bias", | |
| "mean": "decoder1.bn2.running_mean", | |
| "var": "decoder1.bn2.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/decoder1/bn2/BatchNormalization_output_0", | |
| "y": "/decoder1/act_1/Relu_output_0" | |
| }, | |
| { | |
| "op": "conv_t", | |
| "x": "/decoder1/act_1/Relu_output_0", | |
| "y": "/upconv0/ConvTranspose_output_0", | |
| "w": "upconv0.weight", | |
| "b": "upconv0.bias", | |
| "stride": [ | |
| 2, | |
| 2, | |
| 2 | |
| ] | |
| }, | |
| { | |
| "op": "crop", | |
| "x": "/encoder0/act_1/Relu_output_0", | |
| "y": "/cc0/Slice_output_0", | |
| "axis": "d", | |
| "start": 40, | |
| "end_margin": 40 | |
| }, | |
| { | |
| "op": "crop", | |
| "x": "/cc0/Slice_output_0", | |
| "y": "/cc0/Slice_1_output_0", | |
| "axis": "h", | |
| "start": 40, | |
| "end_margin": 40 | |
| }, | |
| { | |
| "op": "crop", | |
| "x": "/cc0/Slice_1_output_0", | |
| "y": "/cc0/Slice_2_output_0", | |
| "axis": "w", | |
| "start": 40, | |
| "end_margin": 40 | |
| }, | |
| { | |
| "op": "concat", | |
| "xs": [ | |
| "/cc0/Slice_2_output_0", | |
| "/upconv0/ConvTranspose_output_0" | |
| ], | |
| "y": "/cc0/Concat_output_0" | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/cc0/Concat_output_0", | |
| "y": "/decoder0/conv1/Conv_output_0", | |
| "w": "decoder0.conv1.weight", | |
| "b": "decoder0.conv1.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/decoder0/conv1/Conv_output_0", | |
| "y": "/decoder0/bn1/BatchNormalization_output_0", | |
| "scale": "decoder0.bn1.weight", | |
| "bias": "decoder0.bn1.bias", | |
| "mean": "decoder0.bn1.running_mean", | |
| "var": "decoder0.bn1.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/decoder0/bn1/BatchNormalization_output_0", | |
| "y": "/decoder0/act/Relu_output_0" | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/decoder0/act/Relu_output_0", | |
| "y": "/decoder0/conv2/Conv_output_0", | |
| "w": "decoder0.conv2.weight", | |
| "b": "decoder0.conv2.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| }, | |
| { | |
| "op": "bn", | |
| "x": "/decoder0/conv2/Conv_output_0", | |
| "y": "/decoder0/bn2/BatchNormalization_output_0", | |
| "scale": "decoder0.bn2.weight", | |
| "bias": "decoder0.bn2.bias", | |
| "mean": "decoder0.bn2.running_mean", | |
| "var": "decoder0.bn2.running_var", | |
| "eps": 9.999999747378752e-06 | |
| }, | |
| { | |
| "op": "relu", | |
| "x": "/decoder0/bn2/BatchNormalization_output_0", | |
| "y": "/decoder0/act_1/Relu_output_0" | |
| }, | |
| { | |
| "op": "conv", | |
| "x": "/decoder0/act_1/Relu_output_0", | |
| "y": "output", | |
| "w": "final_conv.weight", | |
| "b": "final_conv.bias", | |
| "stride": [ | |
| 1, | |
| 1, | |
| 1 | |
| ] | |
| } | |
| ] | |
| } |