Spaces:
Runtime error
Runtime error
Update luciddreamer.py
Browse files- 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
|