| { |
| "format": "cellmap-train-graph/1", |
| "layout": "NDHWC", |
| "input": "input", |
| "output": "output", |
| "context": 88, |
| "params": { |
| "stem.conv1.weight": { |
| "shape": [ |
| 40, |
| 1, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "stem.conv1.bias": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "stem.norm1.weight": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "stem.norm1.bias": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "stem.conv2.weight": { |
| "shape": [ |
| 40, |
| 40, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "stem.conv2.bias": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "stem.norm2.weight": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "stem.norm2.bias": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "stem.skip.weight": { |
| "shape": [ |
| 40, |
| 1, |
| 1, |
| 1, |
| 1 |
| ], |
| "trainable": true |
| }, |
| "stem.skip.bias": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "downs.0.conv.weight": { |
| "shape": [ |
| 80, |
| 40, |
| 2, |
| 2, |
| 2 |
| ], |
| "trainable": true |
| }, |
| "downs.0.conv.bias": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "downs.1.conv.weight": { |
| "shape": [ |
| 160, |
| 80, |
| 2, |
| 2, |
| 2 |
| ], |
| "trainable": true |
| }, |
| "downs.1.conv.bias": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "downs.2.conv.weight": { |
| "shape": [ |
| 320, |
| 160, |
| 2, |
| 2, |
| 2 |
| ], |
| "trainable": true |
| }, |
| "downs.2.conv.bias": { |
| "shape": [ |
| 320 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.0.0.conv1.weight": { |
| "shape": [ |
| 80, |
| 80, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.0.0.conv1.bias": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.0.0.norm1.weight": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.0.0.norm1.bias": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.0.0.conv2.weight": { |
| "shape": [ |
| 80, |
| 80, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.0.0.conv2.bias": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.0.0.norm2.weight": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.0.0.norm2.bias": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.1.0.conv1.weight": { |
| "shape": [ |
| 160, |
| 160, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.1.0.conv1.bias": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.1.0.norm1.weight": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.1.0.norm1.bias": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.1.0.conv2.weight": { |
| "shape": [ |
| 160, |
| 160, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.1.0.conv2.bias": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.1.0.norm2.weight": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.1.0.norm2.bias": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.2.0.conv1.weight": { |
| "shape": [ |
| 320, |
| 320, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.2.0.conv1.bias": { |
| "shape": [ |
| 320 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.2.0.norm1.weight": { |
| "shape": [ |
| 320 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.2.0.norm1.bias": { |
| "shape": [ |
| 320 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.2.0.conv2.weight": { |
| "shape": [ |
| 320, |
| 320, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.2.0.conv2.bias": { |
| "shape": [ |
| 320 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.2.0.norm2.weight": { |
| "shape": [ |
| 320 |
| ], |
| "trainable": true |
| }, |
| "enc_blocks.2.0.norm2.bias": { |
| "shape": [ |
| 320 |
| ], |
| "trainable": true |
| }, |
| "ups.0.reduce.weight": { |
| "shape": [ |
| 160, |
| 320, |
| 1, |
| 1, |
| 1 |
| ], |
| "trainable": true |
| }, |
| "ups.0.reduce.bias": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "ups.1.reduce.weight": { |
| "shape": [ |
| 80, |
| 160, |
| 1, |
| 1, |
| 1 |
| ], |
| "trainable": true |
| }, |
| "ups.1.reduce.bias": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "ups.2.reduce.weight": { |
| "shape": [ |
| 40, |
| 80, |
| 1, |
| 1, |
| 1 |
| ], |
| "trainable": true |
| }, |
| "ups.2.reduce.bias": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.0.0.conv1.weight": { |
| "shape": [ |
| 160, |
| 320, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.0.0.conv1.bias": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.0.0.norm1.weight": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.0.0.norm1.bias": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.0.0.conv2.weight": { |
| "shape": [ |
| 160, |
| 160, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.0.0.conv2.bias": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.0.0.norm2.weight": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.0.0.norm2.bias": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.0.0.skip.weight": { |
| "shape": [ |
| 160, |
| 320, |
| 1, |
| 1, |
| 1 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.0.0.skip.bias": { |
| "shape": [ |
| 160 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.1.0.conv1.weight": { |
| "shape": [ |
| 80, |
| 160, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.1.0.conv1.bias": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.1.0.norm1.weight": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.1.0.norm1.bias": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.1.0.conv2.weight": { |
| "shape": [ |
| 80, |
| 80, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.1.0.conv2.bias": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.1.0.norm2.weight": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.1.0.norm2.bias": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.1.0.skip.weight": { |
| "shape": [ |
| 80, |
| 160, |
| 1, |
| 1, |
| 1 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.1.0.skip.bias": { |
| "shape": [ |
| 80 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.2.0.conv1.weight": { |
| "shape": [ |
| 40, |
| 80, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.2.0.conv1.bias": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.2.0.norm1.weight": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.2.0.norm1.bias": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.2.0.conv2.weight": { |
| "shape": [ |
| 40, |
| 40, |
| 3, |
| 3, |
| 3 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.2.0.conv2.bias": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.2.0.norm2.weight": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.2.0.norm2.bias": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.2.0.skip.weight": { |
| "shape": [ |
| 40, |
| 80, |
| 1, |
| 1, |
| 1 |
| ], |
| "trainable": true |
| }, |
| "dec_blocks.2.0.skip.bias": { |
| "shape": [ |
| 40 |
| ], |
| "trainable": true |
| }, |
| "final_conv.weight": { |
| "shape": [ |
| 1, |
| 40, |
| 1, |
| 1, |
| 1 |
| ], |
| "trainable": true |
| }, |
| "final_conv.bias": { |
| "shape": [ |
| 1 |
| ], |
| "trainable": true |
| } |
| }, |
| "ops": [ |
| { |
| "op": "conv", |
| "x": "input", |
| "y": "/stem/skip/Conv_output_0", |
| "w": "stem.skip.weight", |
| "b": "stem.skip.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "conv", |
| "x": "input", |
| "y": "/stem/conv1/Conv_output_0", |
| "w": "stem.conv1.weight", |
| "b": "stem.conv1.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/stem/conv1/Conv_output_0", |
| "y": "/stem/norm1/InstanceNormalization_output_0", |
| "scale": "stem.norm1.weight", |
| "bias": "stem.norm1.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "relu", |
| "x": "/stem/norm1/InstanceNormalization_output_0", |
| "y": "/stem/act/Relu_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/stem/act/Relu_output_0", |
| "y": "/stem/conv2/Conv_output_0", |
| "w": "stem.conv2.weight", |
| "b": "stem.conv2.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/stem/conv2/Conv_output_0", |
| "y": "/stem/norm2/InstanceNormalization_output_0", |
| "scale": "stem.norm2.weight", |
| "bias": "stem.norm2.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "crop", |
| "x": "/stem/skip/Conv_output_0", |
| "y": "/stem/Slice_output_0", |
| "axis": "d", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/stem/Slice_output_0", |
| "y": "/stem/Slice_1_output_0", |
| "axis": "h", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/stem/Slice_1_output_0", |
| "y": "/stem/Slice_2_output_0", |
| "axis": "w", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "add", |
| "x": "/stem/norm2/InstanceNormalization_output_0", |
| "x2": "/stem/Slice_2_output_0", |
| "y": "/stem/Add_output_0" |
| }, |
| { |
| "op": "relu", |
| "x": "/stem/Add_output_0", |
| "y": "/stem/act_1/Relu_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/stem/act_1/Relu_output_0", |
| "y": "/downs.0/conv/Conv_output_0", |
| "w": "downs.0.conv.weight", |
| "b": "downs.0.conv.bias", |
| "stride": [ |
| 2, |
| 2, |
| 2 |
| ] |
| }, |
| { |
| "op": "conv", |
| "x": "/downs.0/conv/Conv_output_0", |
| "y": "/enc_blocks.0.0/conv1/Conv_output_0", |
| "w": "enc_blocks.0.0.conv1.weight", |
| "b": "enc_blocks.0.0.conv1.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/enc_blocks.0.0/conv1/Conv_output_0", |
| "y": "/enc_blocks.0.0/norm1/InstanceNormalization_output_0", |
| "scale": "enc_blocks.0.0.norm1.weight", |
| "bias": "enc_blocks.0.0.norm1.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "relu", |
| "x": "/enc_blocks.0.0/norm1/InstanceNormalization_output_0", |
| "y": "/enc_blocks.0.0/act/Relu_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/enc_blocks.0.0/act/Relu_output_0", |
| "y": "/enc_blocks.0.0/conv2/Conv_output_0", |
| "w": "enc_blocks.0.0.conv2.weight", |
| "b": "enc_blocks.0.0.conv2.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/enc_blocks.0.0/conv2/Conv_output_0", |
| "y": "/enc_blocks.0.0/norm2/InstanceNormalization_output_0", |
| "scale": "enc_blocks.0.0.norm2.weight", |
| "bias": "enc_blocks.0.0.norm2.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "crop", |
| "x": "/downs.0/conv/Conv_output_0", |
| "y": "/enc_blocks.0.0/Slice_output_0", |
| "axis": "d", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/enc_blocks.0.0/Slice_output_0", |
| "y": "/enc_blocks.0.0/Slice_1_output_0", |
| "axis": "h", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/enc_blocks.0.0/Slice_1_output_0", |
| "y": "/enc_blocks.0.0/Slice_2_output_0", |
| "axis": "w", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "add", |
| "x": "/enc_blocks.0.0/norm2/InstanceNormalization_output_0", |
| "x2": "/enc_blocks.0.0/Slice_2_output_0", |
| "y": "/enc_blocks.0.0/Add_output_0" |
| }, |
| { |
| "op": "relu", |
| "x": "/enc_blocks.0.0/Add_output_0", |
| "y": "/enc_blocks.0.0/act_1/Relu_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/enc_blocks.0.0/act_1/Relu_output_0", |
| "y": "/downs.1/conv/Conv_output_0", |
| "w": "downs.1.conv.weight", |
| "b": "downs.1.conv.bias", |
| "stride": [ |
| 2, |
| 2, |
| 2 |
| ] |
| }, |
| { |
| "op": "conv", |
| "x": "/downs.1/conv/Conv_output_0", |
| "y": "/enc_blocks.1.0/conv1/Conv_output_0", |
| "w": "enc_blocks.1.0.conv1.weight", |
| "b": "enc_blocks.1.0.conv1.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/enc_blocks.1.0/conv1/Conv_output_0", |
| "y": "/enc_blocks.1.0/norm1/InstanceNormalization_output_0", |
| "scale": "enc_blocks.1.0.norm1.weight", |
| "bias": "enc_blocks.1.0.norm1.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "relu", |
| "x": "/enc_blocks.1.0/norm1/InstanceNormalization_output_0", |
| "y": "/enc_blocks.1.0/act/Relu_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/enc_blocks.1.0/act/Relu_output_0", |
| "y": "/enc_blocks.1.0/conv2/Conv_output_0", |
| "w": "enc_blocks.1.0.conv2.weight", |
| "b": "enc_blocks.1.0.conv2.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/enc_blocks.1.0/conv2/Conv_output_0", |
| "y": "/enc_blocks.1.0/norm2/InstanceNormalization_output_0", |
| "scale": "enc_blocks.1.0.norm2.weight", |
| "bias": "enc_blocks.1.0.norm2.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "crop", |
| "x": "/downs.1/conv/Conv_output_0", |
| "y": "/enc_blocks.1.0/Slice_output_0", |
| "axis": "d", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/enc_blocks.1.0/Slice_output_0", |
| "y": "/enc_blocks.1.0/Slice_1_output_0", |
| "axis": "h", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/enc_blocks.1.0/Slice_1_output_0", |
| "y": "/enc_blocks.1.0/Slice_2_output_0", |
| "axis": "w", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "add", |
| "x": "/enc_blocks.1.0/norm2/InstanceNormalization_output_0", |
| "x2": "/enc_blocks.1.0/Slice_2_output_0", |
| "y": "/enc_blocks.1.0/Add_output_0" |
| }, |
| { |
| "op": "relu", |
| "x": "/enc_blocks.1.0/Add_output_0", |
| "y": "/enc_blocks.1.0/act_1/Relu_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/enc_blocks.1.0/act_1/Relu_output_0", |
| "y": "/downs.2/conv/Conv_output_0", |
| "w": "downs.2.conv.weight", |
| "b": "downs.2.conv.bias", |
| "stride": [ |
| 2, |
| 2, |
| 2 |
| ] |
| }, |
| { |
| "op": "conv", |
| "x": "/downs.2/conv/Conv_output_0", |
| "y": "/enc_blocks.2.0/conv1/Conv_output_0", |
| "w": "enc_blocks.2.0.conv1.weight", |
| "b": "enc_blocks.2.0.conv1.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/enc_blocks.2.0/conv1/Conv_output_0", |
| "y": "/enc_blocks.2.0/norm1/InstanceNormalization_output_0", |
| "scale": "enc_blocks.2.0.norm1.weight", |
| "bias": "enc_blocks.2.0.norm1.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "relu", |
| "x": "/enc_blocks.2.0/norm1/InstanceNormalization_output_0", |
| "y": "/enc_blocks.2.0/act/Relu_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/enc_blocks.2.0/act/Relu_output_0", |
| "y": "/enc_blocks.2.0/conv2/Conv_output_0", |
| "w": "enc_blocks.2.0.conv2.weight", |
| "b": "enc_blocks.2.0.conv2.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/enc_blocks.2.0/conv2/Conv_output_0", |
| "y": "/enc_blocks.2.0/norm2/InstanceNormalization_output_0", |
| "scale": "enc_blocks.2.0.norm2.weight", |
| "bias": "enc_blocks.2.0.norm2.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "crop", |
| "x": "/downs.2/conv/Conv_output_0", |
| "y": "/enc_blocks.2.0/Slice_output_0", |
| "axis": "d", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/enc_blocks.2.0/Slice_output_0", |
| "y": "/enc_blocks.2.0/Slice_1_output_0", |
| "axis": "h", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/enc_blocks.2.0/Slice_1_output_0", |
| "y": "/enc_blocks.2.0/Slice_2_output_0", |
| "axis": "w", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "add", |
| "x": "/enc_blocks.2.0/norm2/InstanceNormalization_output_0", |
| "x2": "/enc_blocks.2.0/Slice_2_output_0", |
| "y": "/enc_blocks.2.0/Add_output_0" |
| }, |
| { |
| "op": "relu", |
| "x": "/enc_blocks.2.0/Add_output_0", |
| "y": "/enc_blocks.2.0/act_1/Relu_output_0" |
| }, |
| { |
| "op": "upsample", |
| "x": "/enc_blocks.2.0/act_1/Relu_output_0", |
| "y": "/ups.0/Resize_output_0", |
| "factor": 2 |
| }, |
| { |
| "op": "conv", |
| "x": "/ups.0/Resize_output_0", |
| "y": "/ups.0/reduce/Conv_output_0", |
| "w": "ups.0.reduce.weight", |
| "b": "ups.0.reduce.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "crop", |
| "x": "/enc_blocks.1.0/act_1/Relu_output_0", |
| "y": "/Slice_output_0", |
| "axis": "d", |
| "start": 4, |
| "end_margin": 4 |
| }, |
| { |
| "op": "crop", |
| "x": "/Slice_output_0", |
| "y": "/Slice_1_output_0", |
| "axis": "h", |
| "start": 4, |
| "end_margin": 4 |
| }, |
| { |
| "op": "crop", |
| "x": "/Slice_1_output_0", |
| "y": "/Slice_2_output_0", |
| "axis": "w", |
| "start": 4, |
| "end_margin": 4 |
| }, |
| { |
| "op": "concat", |
| "xs": [ |
| "/ups.0/reduce/Conv_output_0", |
| "/Slice_2_output_0" |
| ], |
| "y": "/Concat_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/Concat_output_0", |
| "y": "/dec_blocks.0.0/skip/Conv_output_0", |
| "w": "dec_blocks.0.0.skip.weight", |
| "b": "dec_blocks.0.0.skip.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "conv", |
| "x": "/Concat_output_0", |
| "y": "/dec_blocks.0.0/conv1/Conv_output_0", |
| "w": "dec_blocks.0.0.conv1.weight", |
| "b": "dec_blocks.0.0.conv1.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/dec_blocks.0.0/conv1/Conv_output_0", |
| "y": "/dec_blocks.0.0/norm1/InstanceNormalization_output_0", |
| "scale": "dec_blocks.0.0.norm1.weight", |
| "bias": "dec_blocks.0.0.norm1.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "relu", |
| "x": "/dec_blocks.0.0/norm1/InstanceNormalization_output_0", |
| "y": "/dec_blocks.0.0/act/Relu_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/dec_blocks.0.0/act/Relu_output_0", |
| "y": "/dec_blocks.0.0/conv2/Conv_output_0", |
| "w": "dec_blocks.0.0.conv2.weight", |
| "b": "dec_blocks.0.0.conv2.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/dec_blocks.0.0/conv2/Conv_output_0", |
| "y": "/dec_blocks.0.0/norm2/InstanceNormalization_output_0", |
| "scale": "dec_blocks.0.0.norm2.weight", |
| "bias": "dec_blocks.0.0.norm2.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "crop", |
| "x": "/dec_blocks.0.0/skip/Conv_output_0", |
| "y": "/dec_blocks.0.0/Slice_output_0", |
| "axis": "d", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/dec_blocks.0.0/Slice_output_0", |
| "y": "/dec_blocks.0.0/Slice_1_output_0", |
| "axis": "h", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/dec_blocks.0.0/Slice_1_output_0", |
| "y": "/dec_blocks.0.0/Slice_2_output_0", |
| "axis": "w", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "add", |
| "x": "/dec_blocks.0.0/norm2/InstanceNormalization_output_0", |
| "x2": "/dec_blocks.0.0/Slice_2_output_0", |
| "y": "/dec_blocks.0.0/Add_output_0" |
| }, |
| { |
| "op": "relu", |
| "x": "/dec_blocks.0.0/Add_output_0", |
| "y": "/dec_blocks.0.0/act_1/Relu_output_0" |
| }, |
| { |
| "op": "upsample", |
| "x": "/dec_blocks.0.0/act_1/Relu_output_0", |
| "y": "/ups.1/Resize_output_0", |
| "factor": 2 |
| }, |
| { |
| "op": "conv", |
| "x": "/ups.1/Resize_output_0", |
| "y": "/ups.1/reduce/Conv_output_0", |
| "w": "ups.1.reduce.weight", |
| "b": "ups.1.reduce.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "crop", |
| "x": "/enc_blocks.0.0/act_1/Relu_output_0", |
| "y": "/Slice_3_output_0", |
| "axis": "d", |
| "start": 16, |
| "end_margin": 16 |
| }, |
| { |
| "op": "crop", |
| "x": "/Slice_3_output_0", |
| "y": "/Slice_4_output_0", |
| "axis": "h", |
| "start": 16, |
| "end_margin": 16 |
| }, |
| { |
| "op": "crop", |
| "x": "/Slice_4_output_0", |
| "y": "/Slice_5_output_0", |
| "axis": "w", |
| "start": 16, |
| "end_margin": 16 |
| }, |
| { |
| "op": "concat", |
| "xs": [ |
| "/ups.1/reduce/Conv_output_0", |
| "/Slice_5_output_0" |
| ], |
| "y": "/Concat_1_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/Concat_1_output_0", |
| "y": "/dec_blocks.1.0/skip/Conv_output_0", |
| "w": "dec_blocks.1.0.skip.weight", |
| "b": "dec_blocks.1.0.skip.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "conv", |
| "x": "/Concat_1_output_0", |
| "y": "/dec_blocks.1.0/conv1/Conv_output_0", |
| "w": "dec_blocks.1.0.conv1.weight", |
| "b": "dec_blocks.1.0.conv1.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/dec_blocks.1.0/conv1/Conv_output_0", |
| "y": "/dec_blocks.1.0/norm1/InstanceNormalization_output_0", |
| "scale": "dec_blocks.1.0.norm1.weight", |
| "bias": "dec_blocks.1.0.norm1.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "relu", |
| "x": "/dec_blocks.1.0/norm1/InstanceNormalization_output_0", |
| "y": "/dec_blocks.1.0/act/Relu_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/dec_blocks.1.0/act/Relu_output_0", |
| "y": "/dec_blocks.1.0/conv2/Conv_output_0", |
| "w": "dec_blocks.1.0.conv2.weight", |
| "b": "dec_blocks.1.0.conv2.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/dec_blocks.1.0/conv2/Conv_output_0", |
| "y": "/dec_blocks.1.0/norm2/InstanceNormalization_output_0", |
| "scale": "dec_blocks.1.0.norm2.weight", |
| "bias": "dec_blocks.1.0.norm2.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "crop", |
| "x": "/dec_blocks.1.0/skip/Conv_output_0", |
| "y": "/dec_blocks.1.0/Slice_output_0", |
| "axis": "d", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/dec_blocks.1.0/Slice_output_0", |
| "y": "/dec_blocks.1.0/Slice_1_output_0", |
| "axis": "h", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/dec_blocks.1.0/Slice_1_output_0", |
| "y": "/dec_blocks.1.0/Slice_2_output_0", |
| "axis": "w", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "add", |
| "x": "/dec_blocks.1.0/norm2/InstanceNormalization_output_0", |
| "x2": "/dec_blocks.1.0/Slice_2_output_0", |
| "y": "/dec_blocks.1.0/Add_output_0" |
| }, |
| { |
| "op": "relu", |
| "x": "/dec_blocks.1.0/Add_output_0", |
| "y": "/dec_blocks.1.0/act_1/Relu_output_0" |
| }, |
| { |
| "op": "upsample", |
| "x": "/dec_blocks.1.0/act_1/Relu_output_0", |
| "y": "/ups.2/Resize_output_0", |
| "factor": 2 |
| }, |
| { |
| "op": "conv", |
| "x": "/ups.2/Resize_output_0", |
| "y": "/ups.2/reduce/Conv_output_0", |
| "w": "ups.2.reduce.weight", |
| "b": "ups.2.reduce.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "crop", |
| "x": "/stem/act_1/Relu_output_0", |
| "y": "/Slice_6_output_0", |
| "axis": "d", |
| "start": 40, |
| "end_margin": 40 |
| }, |
| { |
| "op": "crop", |
| "x": "/Slice_6_output_0", |
| "y": "/Slice_7_output_0", |
| "axis": "h", |
| "start": 40, |
| "end_margin": 40 |
| }, |
| { |
| "op": "crop", |
| "x": "/Slice_7_output_0", |
| "y": "/Slice_8_output_0", |
| "axis": "w", |
| "start": 40, |
| "end_margin": 40 |
| }, |
| { |
| "op": "concat", |
| "xs": [ |
| "/ups.2/reduce/Conv_output_0", |
| "/Slice_8_output_0" |
| ], |
| "y": "/Concat_2_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/Concat_2_output_0", |
| "y": "/dec_blocks.2.0/skip/Conv_output_0", |
| "w": "dec_blocks.2.0.skip.weight", |
| "b": "dec_blocks.2.0.skip.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "conv", |
| "x": "/Concat_2_output_0", |
| "y": "/dec_blocks.2.0/conv1/Conv_output_0", |
| "w": "dec_blocks.2.0.conv1.weight", |
| "b": "dec_blocks.2.0.conv1.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/dec_blocks.2.0/conv1/Conv_output_0", |
| "y": "/dec_blocks.2.0/norm1/InstanceNormalization_output_0", |
| "scale": "dec_blocks.2.0.norm1.weight", |
| "bias": "dec_blocks.2.0.norm1.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "relu", |
| "x": "/dec_blocks.2.0/norm1/InstanceNormalization_output_0", |
| "y": "/dec_blocks.2.0/act/Relu_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/dec_blocks.2.0/act/Relu_output_0", |
| "y": "/dec_blocks.2.0/conv2/Conv_output_0", |
| "w": "dec_blocks.2.0.conv2.weight", |
| "b": "dec_blocks.2.0.conv2.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| }, |
| { |
| "op": "inorm", |
| "x": "/dec_blocks.2.0/conv2/Conv_output_0", |
| "y": "/dec_blocks.2.0/norm2/InstanceNormalization_output_0", |
| "scale": "dec_blocks.2.0.norm2.weight", |
| "bias": "dec_blocks.2.0.norm2.bias", |
| "eps": 9.999999747378752e-06 |
| }, |
| { |
| "op": "crop", |
| "x": "/dec_blocks.2.0/skip/Conv_output_0", |
| "y": "/dec_blocks.2.0/Slice_output_0", |
| "axis": "d", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/dec_blocks.2.0/Slice_output_0", |
| "y": "/dec_blocks.2.0/Slice_1_output_0", |
| "axis": "h", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "crop", |
| "x": "/dec_blocks.2.0/Slice_1_output_0", |
| "y": "/dec_blocks.2.0/Slice_2_output_0", |
| "axis": "w", |
| "start": 2, |
| "end_margin": 2 |
| }, |
| { |
| "op": "add", |
| "x": "/dec_blocks.2.0/norm2/InstanceNormalization_output_0", |
| "x2": "/dec_blocks.2.0/Slice_2_output_0", |
| "y": "/dec_blocks.2.0/Add_output_0" |
| }, |
| { |
| "op": "relu", |
| "x": "/dec_blocks.2.0/Add_output_0", |
| "y": "/dec_blocks.2.0/act_1/Relu_output_0" |
| }, |
| { |
| "op": "conv", |
| "x": "/dec_blocks.2.0/act_1/Relu_output_0", |
| "y": "output", |
| "w": "final_conv.weight", |
| "b": "final_conv.bias", |
| "stride": [ |
| 1, |
| 1, |
| 1 |
| ] |
| } |
| ] |
| } |
