Monai, Medical Imaging, Pytorch, Triton Inference Server, Torchserve.
ModelHandler from ts.torch_handler.unit_tests.test_utils.mock_context import MockContext from pathlib import Path import os MODEL_PT_FILE='traced_segres_model.pt' CURR_FILE_PATH = Path(__file__).parent.absolute() EXAMPLE_ROOT_DIR=CURR_FILE_PATH TEST_DATA=os.path.join(CURR_FILE_PATH, 'test.nii.gz')