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