Jie Hu
commited on
Commit
·
9c25e98
1
Parent(s):
23c29a5
init project
Browse files- .DS_Store +0 -0
- app.py +0 -1
- modules/.DS_Store +0 -0
- modules/dust3r/cloud_opt/base_opt.py +3 -0
.DS_Store
CHANGED
|
Binary files a/.DS_Store and b/.DS_Store differ
|
|
|
app.py
CHANGED
|
@@ -85,7 +85,6 @@ def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world,
|
|
| 85 |
scene.export(file_obj=outfile)
|
| 86 |
return outfile
|
| 87 |
|
| 88 |
-
# @spaces.GPU(duration=180)
|
| 89 |
def get_3D_model_from_scene(outdir, scene, min_conf_thr=3, as_pointcloud=False, mask_sky=False,
|
| 90 |
clean_depth=False, transparent_cams=False, cam_size=0.05):
|
| 91 |
"""
|
|
|
|
| 85 |
scene.export(file_obj=outfile)
|
| 86 |
return outfile
|
| 87 |
|
|
|
|
| 88 |
def get_3D_model_from_scene(outdir, scene, min_conf_thr=3, as_pointcloud=False, mask_sky=False,
|
| 89 |
clean_depth=False, transparent_cams=False, cam_size=0.05):
|
| 90 |
"""
|
modules/.DS_Store
CHANGED
|
Binary files a/modules/.DS_Store and b/modules/.DS_Store differ
|
|
|
modules/dust3r/cloud_opt/base_opt.py
CHANGED
|
@@ -55,6 +55,9 @@ class BasePCOptimizer (nn.Module):
|
|
| 55 |
iterationsCount=None,
|
| 56 |
verbose=True):
|
| 57 |
super().__init__()
|
|
|
|
|
|
|
|
|
|
| 58 |
if not isinstance(view1['idx'], list):
|
| 59 |
view1['idx'] = view1['idx'].tolist()
|
| 60 |
if not isinstance(view2['idx'], list):
|
|
|
|
| 55 |
iterationsCount=None,
|
| 56 |
verbose=True):
|
| 57 |
super().__init__()
|
| 58 |
+
|
| 59 |
+
self.device = device
|
| 60 |
+
|
| 61 |
if not isinstance(view1['idx'], list):
|
| 62 |
view1['idx'] = view1['idx'].tolist()
|
| 63 |
if not isinstance(view2['idx'], list):
|