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
|