Diff of /utils/config.py [000000] .. [4df946]

Switch to side-by-side view

--- a
+++ b/utils/config.py
@@ -0,0 +1,25 @@
+import os
+
+INFERENCE_GPUS = [8]
+CENTER_SIZE = 2000  # The effective patch size.
+BORDER_SIZE = 100  # The boarder size containing surrounding information.
+PATCH_SIZE = CENTER_SIZE + 2 * BORDER_SIZE
+format_mapping = {
+    'tif': 'OpenSlide',
+    'svs': 'OpenSlide',
+    'ndpi': 'OpenSlide',
+    'scn': 'OpenSlide',
+    'mrxs': 'OpenSlide',
+    'bif': 'OpenSlide',
+    'vms': 'OpenSlide',
+}
+MODEL_NAME = 'stomach'
+INPUT_KEY = 'output'
+PREDICT_KEY = 'output'
+TF_SERVING_HOST = os.environ.get('TF_SERVING_HOST', '127.0.0.1')
+TF_SERVING_PORT = int(os.environ.get('TF_SERVING_PORT', 9000))
+TEMP_DIR = './temp/'  # Where to save the predicted patches.
+THUMBNAIL_RATIO = 10  # Down-sample ratio from the predictions to the thumbnail.
+KEEP_TEMP = False  # Whether to keep the predicted patches after generated the thumbnail.
+DO_POST_PROCESSING = False  # Whether to post-process the predicted patches.
+FILTER_KERNEL = 9  # The kernel size for post-processing.