Monai, Medical Imaging, Pytorch, Triton Inference Server, Torchserve.
the expected output (image). In this particular case, we are demonstrating a segmentation model and so the input and output dims are the same. More information about the config.pbtxt file is available at the official Github for Triton: