config.pbtxt

backend: "python"
max_batch_size: 0  # Keep batching disabled as per original config

input [
  {
    name: "input_jp2_bytes"  # New input name for JP2 bytes
    data_type: TYPE_STRING  # Use TYPE_STRING for bytes
    dims: [ 3 ]             # Expecting 3 elements: Red, Green, NIR bytes
  }
]

output [
  {
    name: "output_mask"
    data_type: TYPE_UINT8
    dims: [-1, -1]     # Variable height, width
  }
]

# Optional: Specify instance_group if running on GPU
# instance_group [
#   {
#     kind: KIND_GPU
#   }
# ]