andreped commited on
Commit
f26565c
1 Parent(s): c58ed05

Fixed camera position of Model3D renderer

Browse files
Files changed (1) hide show
  1. demo/src/gui.py +10 -8
demo/src/gui.py CHANGED
@@ -55,8 +55,10 @@ class WebUI:
55
  self.volume_renderer = gr.Model3D(
56
  clear_color=[0.0, 0.0, 0.0, 0.0],
57
  label="3D Model",
 
58
  visible=True,
59
  elem_id="model-3d",
 
60
  ).style(height=512)
61
 
62
  def set_class_name(self, value):
@@ -73,13 +75,13 @@ class WebUI:
73
 
74
  def process(self, mesh_file_name):
75
  path = mesh_file_name.name
76
- run_model(
77
- path,
78
- model_path=os.path.join(self.cwd, "resources/models/"),
79
- task=self.class_names[self.class_name],
80
- name=self.result_names[self.class_name],
81
- )
82
- LOGGER.info("Converting prediction NIfTI to GLB...")
83
  nifti_to_glb("prediction.nii.gz")
84
 
85
  LOGGER.info("Loading CT to numpy...")
@@ -195,7 +197,7 @@ class WebUI:
195
 
196
  gr.Markdown(
197
  """
198
- **NOTE:** Inference might take several minutes (ETA: ~8 minutes), see logs to the left. \\
199
  The segmentation will be available in the 2D and 3D viewers below when finished.
200
  """
201
  )
 
55
  self.volume_renderer = gr.Model3D(
56
  clear_color=[0.0, 0.0, 0.0, 0.0],
57
  label="3D Model",
58
+ show_label=True,
59
  visible=True,
60
  elem_id="model-3d",
61
+ camera_position=[90, 180, 768],
62
  ).style(height=512)
63
 
64
  def set_class_name(self, value):
 
75
 
76
  def process(self, mesh_file_name):
77
  path = mesh_file_name.name
78
+ #run_model(
79
+ # path,
80
+ # model_path=os.path.join(self.cwd, "resources/models/"),
81
+ # task=self.class_names[self.class_name],
82
+ # name=self.result_names[self.class_name],
83
+ #)
84
+ LOGGER.info("Converting prediction NIfTI to OBJ...")
85
  nifti_to_glb("prediction.nii.gz")
86
 
87
  LOGGER.info("Loading CT to numpy...")
 
197
 
198
  gr.Markdown(
199
  """
200
+ **NOTE:** Inference might take several minutes (Airways: ~8 minutes), see logs to the left. \\
201
  The segmentation will be available in the 2D and 3D viewers below when finished.
202
  """
203
  )