diff --git a/vision/medical_imaging/3d-unet-kits19/base_SUT.py b/vision/medical_imaging/3d-unet-kits19/base_SUT.py index d94319b773..9921e31dc3 100644 --- a/vision/medical_imaging/3d-unet-kits19/base_SUT.py +++ b/vision/medical_imaging/3d-unet-kits19/base_SUT.py @@ -23,7 +23,7 @@ import inference_utils as infu from kits_QSL import get_kits_QSL -from global_vars import * +from global_vars import ROI_SHAPE, SLIDE_OVERLAP_FACTOR class BASE_3DUNET_SUT: diff --git a/vision/medical_imaging/3d-unet-kits19/inference_utils.py b/vision/medical_imaging/3d-unet-kits19/inference_utils.py index 333f9846cf..4514c70767 100644 --- a/vision/medical_imaging/3d-unet-kits19/inference_utils.py +++ b/vision/medical_imaging/3d-unet-kits19/inference_utils.py @@ -20,7 +20,7 @@ import time from scipy import signal -from global_vars import * +from global_vars import ROI_SHAPE, SLIDE_OVERLAP_FACTOR __doc__ = """ Collection of utilities 3D UNet MLPerf-Inference reference model uses. diff --git a/vision/medical_imaging/3d-unet-kits19/preprocess.py b/vision/medical_imaging/3d-unet-kits19/preprocess.py index ce7ea33c49..76f6a88c79 100644 --- a/vision/medical_imaging/3d-unet-kits19/preprocess.py +++ b/vision/medical_imaging/3d-unet-kits19/preprocess.py @@ -29,7 +29,7 @@ from scipy.ndimage.interpolation import zoom from pathlib import Path -from global_vars import * +from global_vars import CALIB_CASES, TARGET_CASES, MEAN_VAL, STDDEV_VAL, MIN_CLIP_VAL, MAX_CLIP_VAL, PADDING_VAL, TARGET_SPACING, ROI_SHAPE, SLIDE_OVERLAP_FACTOR, ROI_SHAPE, CHECKSUM_CALIB_FILE, CHECKSUM_INFER_FILE __doc__ = """ diff --git a/vision/medical_imaging/3d-unet-kits19/unet_pytorch_to_onnx.py b/vision/medical_imaging/3d-unet-kits19/unet_pytorch_to_onnx.py index 8164d7d05c..c5394b05e1 100644 --- a/vision/medical_imaging/3d-unet-kits19/unet_pytorch_to_onnx.py +++ b/vision/medical_imaging/3d-unet-kits19/unet_pytorch_to_onnx.py @@ -16,7 +16,7 @@ # limitations under the License. -from global_vars import * +from global_vars import ROI_SHAPE from pathlib import Path import torch import argparse