Added Room Layout results

montescene · Apr 22, 2021 · fa43311 · fa43311
1 parent 00e6e73
commit fa43311
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/demo.py b/demo.py
@@ -13,8 +13,8 @@
 
 flags.DEFINE_string('scene', 'scene0000_00', 'scene ID')
 flags.DEFINE_string('shapenet_dir', '/media/shreyas/4aa82be1-14a8-47f7-93a7-171e3ebac2b0/Datasets/ShapeNetCore.v2', 'shapenet dir')
-flags.DEFINE_bool('add_objects', False, 'add objects to the scene')
-flags.DEFINE_bool('add_layout', False, 'add room layout to the scene')
+flags.DEFINE_bool('add_objects', True, 'add objects to the scene')
+flags.DEFINE_bool('add_layout', True, 'add room layout to the scene')
 
 def visualize(sceneID, add_objects=True, add_layout=True):
  # Visualize
@@ -34,7 +34,6 @@ def visualize(sceneID, add_objects=True, add_layout=True):
  col = [0, 0, 1]
  lineSets = drawOpen3dCylLines([obj.orientedBB], col)
  lineGeometryList.append(lineSets)
-
  finalMesh = deepcopy(mctsObjs.sceneMesh)
  for i in range(0, len(geometryList)):
  finalMesh += lineGeometryList[i]