{ "encoder0.conv1.convs.0.weight": { "torch": "encoder0.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "encoder0.conv1.convs.0.bias": { "torch": "encoder0.conv1.bias", "transform": "identity", "to_onnx": [] }, "encoder0.conv1.convs.1.weight": { "torch": "encoder0.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "encoder0.conv1.convs.2.weight": { "torch": "encoder0.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "encoder0.bn1.weight": { "torch": "encoder0.bn1.weight", "transform": "identity", "to_onnx": [] }, "encoder0.bn1.bias": { "torch": "encoder0.bn1.bias", "transform": "identity", "to_onnx": [] }, "encoder0.bn1.running_mean": { "torch": "encoder0.bn1.running_mean", "transform": "identity", "to_onnx": [] }, "encoder0.bn1.running_var": { "torch": "encoder0.bn1.running_var", "transform": "identity", "to_onnx": [] }, "encoder0.conv2.convs.0.weight": { "torch": "encoder0.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "encoder0.conv2.convs.0.bias": { "torch": "encoder0.conv2.bias", "transform": "identity", "to_onnx": [] }, "encoder0.conv2.convs.1.weight": { "torch": "encoder0.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "encoder0.conv2.convs.2.weight": { "torch": "encoder0.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "encoder0.bn2.weight": { "torch": "encoder0.bn2.weight", "transform": "identity", "to_onnx": [] }, "encoder0.bn2.bias": { "torch": "encoder0.bn2.bias", "transform": "identity", "to_onnx": [] }, "encoder0.bn2.running_mean": { "torch": "encoder0.bn2.running_mean", "transform": "identity", "to_onnx": [] }, "encoder0.bn2.running_var": { "torch": "encoder0.bn2.running_var", "transform": "identity", "to_onnx": [] }, "encoder1.conv1.convs.0.weight": { "torch": "encoder1.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "encoder1.conv1.convs.0.bias": { "torch": "encoder1.conv1.bias", "transform": "identity", "to_onnx": [] }, "encoder1.conv1.convs.1.weight": { "torch": "encoder1.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "encoder1.conv1.convs.2.weight": { "torch": "encoder1.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "encoder1.bn1.weight": { "torch": "encoder1.bn1.weight", "transform": "identity", "to_onnx": [] }, "encoder1.bn1.bias": { "torch": "encoder1.bn1.bias", "transform": "identity", "to_onnx": [] }, "encoder1.bn1.running_mean": { "torch": "encoder1.bn1.running_mean", "transform": "identity", "to_onnx": [] }, "encoder1.bn1.running_var": { "torch": "encoder1.bn1.running_var", "transform": "identity", "to_onnx": [] }, "encoder1.conv2.convs.0.weight": { "torch": "encoder1.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "encoder1.conv2.convs.0.bias": { "torch": "encoder1.conv2.bias", "transform": "identity", "to_onnx": [] }, "encoder1.conv2.convs.1.weight": { "torch": "encoder1.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "encoder1.conv2.convs.2.weight": { "torch": "encoder1.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "encoder1.bn2.weight": { "torch": "encoder1.bn2.weight", "transform": "identity", "to_onnx": [] }, "encoder1.bn2.bias": { "torch": "encoder1.bn2.bias", "transform": "identity", "to_onnx": [] }, "encoder1.bn2.running_mean": { "torch": "encoder1.bn2.running_mean", "transform": "identity", "to_onnx": [] }, "encoder1.bn2.running_var": { "torch": "encoder1.bn2.running_var", "transform": "identity", "to_onnx": [] }, "encoder2.conv1.convs.0.weight": { "torch": "encoder2.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "encoder2.conv1.convs.0.bias": { "torch": "encoder2.conv1.bias", "transform": "identity", "to_onnx": [] }, "encoder2.conv1.convs.1.weight": { "torch": "encoder2.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "encoder2.conv1.convs.2.weight": { "torch": "encoder2.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "encoder2.bn1.weight": { "torch": "encoder2.bn1.weight", "transform": "identity", "to_onnx": [] }, "encoder2.bn1.bias": { "torch": "encoder2.bn1.bias", "transform": "identity", "to_onnx": [] }, "encoder2.bn1.running_mean": { "torch": "encoder2.bn1.running_mean", "transform": "identity", "to_onnx": [] }, "encoder2.bn1.running_var": { "torch": "encoder2.bn1.running_var", "transform": "identity", "to_onnx": [] }, "encoder2.conv2.convs.0.weight": { "torch": "encoder2.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "encoder2.conv2.convs.0.bias": { "torch": "encoder2.conv2.bias", "transform": "identity", "to_onnx": [] }, "encoder2.conv2.convs.1.weight": { "torch": "encoder2.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "encoder2.conv2.convs.2.weight": { "torch": "encoder2.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "encoder2.bn2.weight": { "torch": "encoder2.bn2.weight", "transform": "identity", "to_onnx": [] }, "encoder2.bn2.bias": { "torch": "encoder2.bn2.bias", "transform": "identity", "to_onnx": [] }, "encoder2.bn2.running_mean": { "torch": "encoder2.bn2.running_mean", "transform": "identity", "to_onnx": [] }, "encoder2.bn2.running_var": { "torch": "encoder2.bn2.running_var", "transform": "identity", "to_onnx": [] }, "bottleneck.conv1.convs.0.weight": { "torch": "bottleneck.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "bottleneck.conv1.convs.0.bias": { "torch": "bottleneck.conv1.bias", "transform": "identity", "to_onnx": [] }, "bottleneck.conv1.convs.1.weight": { "torch": "bottleneck.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "bottleneck.conv1.convs.2.weight": { "torch": "bottleneck.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "bottleneck.bn1.weight": { "torch": "bottleneck.bn1.weight", "transform": "identity", "to_onnx": [] }, "bottleneck.bn1.bias": { "torch": "bottleneck.bn1.bias", "transform": "identity", "to_onnx": [] }, "bottleneck.bn1.running_mean": { "torch": "bottleneck.bn1.running_mean", "transform": "identity", "to_onnx": [] }, "bottleneck.bn1.running_var": { "torch": "bottleneck.bn1.running_var", "transform": "identity", "to_onnx": [] }, "bottleneck.conv2.convs.0.weight": { "torch": "bottleneck.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "bottleneck.conv2.convs.0.bias": { "torch": "bottleneck.conv2.bias", "transform": "identity", "to_onnx": [] }, "bottleneck.conv2.convs.1.weight": { "torch": "bottleneck.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "bottleneck.conv2.convs.2.weight": { "torch": "bottleneck.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "bottleneck.bn2.weight": { "torch": "bottleneck.bn2.weight", "transform": "identity", "to_onnx": [] }, "bottleneck.bn2.bias": { "torch": "bottleneck.bn2.bias", "transform": "identity", "to_onnx": [] }, "bottleneck.bn2.running_mean": { "torch": "bottleneck.bn2.running_mean", "transform": "identity", "to_onnx": [] }, "bottleneck.bn2.running_var": { "torch": "bottleneck.bn2.running_var", "transform": "identity", "to_onnx": [] }, "upconv2.conv.convs.0.weight": { "torch": "upconv2.weight", "transform": "permute(1,2,3,4,0).reshape(Cout*8, Cin, 1, 1)", "to_onnx": [ { "permute": [ 1, 2, 3, 4, 0 ] }, { "reshape": [ 1024, 256, 1, 1 ] } ] }, "upconv2.conv.convs.0.bias": { "torch": "upconv2.bias", "transform": "repeat_interleave(8)", "to_onnx": [ { "repeat_interleave": [ 0, 8 ] } ] }, "decoder2.conv1.convs.0.weight": { "torch": "decoder2.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "decoder2.conv1.convs.0.bias": { "torch": "decoder2.conv1.bias", "transform": "identity", "to_onnx": [] }, "decoder2.conv1.convs.1.weight": { "torch": "decoder2.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "decoder2.conv1.convs.2.weight": { "torch": "decoder2.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "decoder2.bn1.weight": { "torch": "decoder2.bn1.weight", "transform": "identity", "to_onnx": [] }, "decoder2.bn1.bias": { "torch": "decoder2.bn1.bias", "transform": "identity", "to_onnx": [] }, "decoder2.bn1.running_mean": { "torch": "decoder2.bn1.running_mean", "transform": "identity", "to_onnx": [] }, "decoder2.bn1.running_var": { "torch": "decoder2.bn1.running_var", "transform": "identity", "to_onnx": [] }, "decoder2.conv2.convs.0.weight": { "torch": "decoder2.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "decoder2.conv2.convs.0.bias": { "torch": "decoder2.conv2.bias", "transform": "identity", "to_onnx": [] }, "decoder2.conv2.convs.1.weight": { "torch": "decoder2.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "decoder2.conv2.convs.2.weight": { "torch": "decoder2.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "decoder2.bn2.weight": { "torch": "decoder2.bn2.weight", "transform": "identity", "to_onnx": [] }, "decoder2.bn2.bias": { "torch": "decoder2.bn2.bias", "transform": "identity", "to_onnx": [] }, "decoder2.bn2.running_mean": { "torch": "decoder2.bn2.running_mean", "transform": "identity", "to_onnx": [] }, "decoder2.bn2.running_var": { "torch": "decoder2.bn2.running_var", "transform": "identity", "to_onnx": [] }, "upconv1.conv.convs.0.weight": { "torch": "upconv1.weight", "transform": "permute(1,2,3,4,0).reshape(Cout*8, Cin, 1, 1)", "to_onnx": [ { "permute": [ 1, 2, 3, 4, 0 ] }, { "reshape": [ 512, 128, 1, 1 ] } ] }, "upconv1.conv.convs.0.bias": { "torch": "upconv1.bias", "transform": "repeat_interleave(8)", "to_onnx": [ { "repeat_interleave": [ 0, 8 ] } ] }, "decoder1.conv1.convs.0.weight": { "torch": "decoder1.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "decoder1.conv1.convs.0.bias": { "torch": "decoder1.conv1.bias", "transform": "identity", "to_onnx": [] }, "decoder1.conv1.convs.1.weight": { "torch": "decoder1.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "decoder1.conv1.convs.2.weight": { "torch": "decoder1.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "decoder1.bn1.weight": { "torch": "decoder1.bn1.weight", "transform": "identity", "to_onnx": [] }, "decoder1.bn1.bias": { "torch": "decoder1.bn1.bias", "transform": "identity", "to_onnx": [] }, "decoder1.bn1.running_mean": { "torch": "decoder1.bn1.running_mean", "transform": "identity", "to_onnx": [] }, "decoder1.bn1.running_var": { "torch": "decoder1.bn1.running_var", "transform": "identity", "to_onnx": [] }, "decoder1.conv2.convs.0.weight": { "torch": "decoder1.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "decoder1.conv2.convs.0.bias": { "torch": "decoder1.conv2.bias", "transform": "identity", "to_onnx": [] }, "decoder1.conv2.convs.1.weight": { "torch": "decoder1.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "decoder1.conv2.convs.2.weight": { "torch": "decoder1.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "decoder1.bn2.weight": { "torch": "decoder1.bn2.weight", "transform": "identity", "to_onnx": [] }, "decoder1.bn2.bias": { "torch": "decoder1.bn2.bias", "transform": "identity", "to_onnx": [] }, "decoder1.bn2.running_mean": { "torch": "decoder1.bn2.running_mean", "transform": "identity", "to_onnx": [] }, "decoder1.bn2.running_var": { "torch": "decoder1.bn2.running_var", "transform": "identity", "to_onnx": [] }, "upconv0.conv.convs.0.weight": { "torch": "upconv0.weight", "transform": "permute(1,2,3,4,0).reshape(Cout*8, Cin, 1, 1)", "to_onnx": [ { "permute": [ 1, 2, 3, 4, 0 ] }, { "reshape": [ 256, 64, 1, 1 ] } ] }, "upconv0.conv.convs.0.bias": { "torch": "upconv0.bias", "transform": "repeat_interleave(8)", "to_onnx": [ { "repeat_interleave": [ 0, 8 ] } ] }, "decoder0.conv1.convs.0.weight": { "torch": "decoder0.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "decoder0.conv1.convs.0.bias": { "torch": "decoder0.conv1.bias", "transform": "identity", "to_onnx": [] }, "decoder0.conv1.convs.1.weight": { "torch": "decoder0.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "decoder0.conv1.convs.2.weight": { "torch": "decoder0.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "decoder0.bn1.weight": { "torch": "decoder0.bn1.weight", "transform": "identity", "to_onnx": [] }, "decoder0.bn1.bias": { "torch": "decoder0.bn1.bias", "transform": "identity", "to_onnx": [] }, "decoder0.bn1.running_mean": { "torch": "decoder0.bn1.running_mean", "transform": "identity", "to_onnx": [] }, "decoder0.bn1.running_var": { "torch": "decoder0.bn1.running_var", "transform": "identity", "to_onnx": [] }, "decoder0.conv2.convs.0.weight": { "torch": "decoder0.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "decoder0.conv2.convs.0.bias": { "torch": "decoder0.conv2.bias", "transform": "identity", "to_onnx": [] }, "decoder0.conv2.convs.1.weight": { "torch": "decoder0.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "decoder0.conv2.convs.2.weight": { "torch": "decoder0.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "decoder0.bn2.weight": { "torch": "decoder0.bn2.weight", "transform": "identity", "to_onnx": [] }, "decoder0.bn2.bias": { "torch": "decoder0.bn2.bias", "transform": "identity", "to_onnx": [] }, "decoder0.bn2.running_mean": { "torch": "decoder0.bn2.running_mean", "transform": "identity", "to_onnx": [] }, "decoder0.bn2.running_var": { "torch": "decoder0.bn2.running_var", "transform": "identity", "to_onnx": [] }, "final_conv.convs.0.weight": { "torch": "final_conv.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "final_conv.convs.0.bias": { "torch": "final_conv.bias", "transform": "identity", "to_onnx": [] } }