svjack commited on
Commit
347b551
·
verified ·
1 Parent(s): 19c4516

Update luciddreamer.py

Browse files
Files changed (1) hide show
  1. luciddreamer.py +4 -1
luciddreamer.py CHANGED
@@ -202,11 +202,14 @@ class LucidDreamer:
202
  return outfile
203
 
204
  def save_ply(self, fpath=None):
 
 
205
  if fpath is None:
206
  dpath = os.path.join(self.root, self.timestamp)
207
  fpath = os.path.join(dpath, 'gsplat.ply')
208
  os.makedirs(dpath, exist_ok=True)
209
  if not os.path.exists(fpath):
 
210
  self.gaussians.save_ply(fpath)
211
  else:
212
  self.gaussians.load_ply(fpath)
@@ -573,4 +576,4 @@ class LucidDreamer:
573
  })
574
 
575
  progress(1, desc='[3/4] Baking Gaussians...')
576
- return traindata
 
202
  return outfile
203
 
204
  def save_ply(self, fpath=None):
205
+ dpath = os.path.join(self.root, self.timestamp)
206
+ os.makedirs(dpath, exist_ok=True)
207
  if fpath is None:
208
  dpath = os.path.join(self.root, self.timestamp)
209
  fpath = os.path.join(dpath, 'gsplat.ply')
210
  os.makedirs(dpath, exist_ok=True)
211
  if not os.path.exists(fpath):
212
+ os.makedirs(fpath, exist_ok=True)
213
  self.gaussians.save_ply(fpath)
214
  else:
215
  self.gaussians.load_ply(fpath)
 
576
  })
577
 
578
  progress(1, desc='[3/4] Baking Gaussians...')
579
+ return traindata