Skip to content

Commit 9d135bc

Browse files
Specify wildcard imports in medical imaging task (#2521)
* Add shebangs in bulk pre-testing * Update to exec mode torebase * Revert "Update to exec mode" This reverts commit 3099c81. * Revert "Add shebangs in bulk pre-testing" This reverts commit 0ee8fb9. * Update wildcards for medical imaging --------- Co-authored-by: Pablo Gonzalez <pgmpablo97@gmail.com>
1 parent 3a3111a commit 9d135bc

4 files changed

Lines changed: 4 additions & 4 deletions

File tree

vision/medical_imaging/3d-unet-kits19/base_SUT.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
import inference_utils as infu
2424

2525
from kits_QSL import get_kits_QSL
26-
from global_vars import *
26+
from global_vars import ROI_SHAPE, SLIDE_OVERLAP_FACTOR
2727

2828

2929
class BASE_3DUNET_SUT:

vision/medical_imaging/3d-unet-kits19/inference_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
import time
2121
from scipy import signal
2222

23-
from global_vars import *
23+
from global_vars import ROI_SHAPE, SLIDE_OVERLAP_FACTOR
2424

2525
__doc__ = """
2626
Collection of utilities 3D UNet MLPerf-Inference reference model uses.

vision/medical_imaging/3d-unet-kits19/preprocess.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
from scipy.ndimage.interpolation import zoom
3030
from pathlib import Path
3131

32-
from global_vars import *
32+
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
3333

3434

3535
__doc__ = """

vision/medical_imaging/3d-unet-kits19/unet_pytorch_to_onnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
# limitations under the License.
1717

1818

19-
from global_vars import *
19+
from global_vars import ROI_SHAPE
2020
from pathlib import Path
2121
import torch
2222
import argparse

0 commit comments

Comments
 (0)