Minor fix

snap-stanford · Mar 13, 2023 · d9ddebb · d9ddebb
1 parent 3f1806c
commit d9ddebb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/inference_zero_shot.ipynb b/inference_zero_shot.ipynb
@@ -2342,6 +2342,10 @@
     "        \"n_examples\": 200,\n",
     "        \"num_processes_3d\": 5,\n",
     "        \"n_queries_per_class\": 1,\n",
+    "        \"color_map_3d\": \"same\",\n",
+    "        \"add_thick_surf\": (0, 0.5),\n",
+    "        \"add_thick_depth\": (0, 0.5),\n",
+    "        \"image_size_3d\": (256, 256),\n",
     "        # 2D examples\n",
     "        \"seed\": 102,\n",
     "        \"use_seed_2d\": True,\n",

diff --git a/inference_zero_shot.py b/inference_zero_shot.py
@@ -2282,6 +2282,10 @@ def parse_selector_from_image(
         "n_examples": 200,
         "num_processes_3d": 5,
         "n_queries_per_class": 1,
+        "color_map_3d": "same",
+        "add_thick_surf": (0, 0.5),
+        "add_thick_depth": (0, 0.5),
+        "image_size_3d": (256, 256),
         # 2D examples
         "seed": 102,
         "use_seed_2d": True,