Update mast3r.py
Browse files
mast3r.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
#
|
2 |
import torch
|
3 |
from mini_dust3r.api import OptimizedResult, inferece_dust3r, log_optimized_result
|
4 |
from mini_dust3r.model import AsymmetricCroCo3DStereo
|
@@ -8,7 +8,7 @@ from mini_dust3r.utils.misc import (
|
|
8 |
from mini_dust3r.model import load_model
|
9 |
from pathlib import Path
|
10 |
import uuid
|
11 |
-
from
|
12 |
import rerun as rr
|
13 |
import rerun.blueprint as rrb
|
14 |
import os
|
@@ -113,7 +113,6 @@ class MASt3R:
|
|
113 |
)
|
114 |
return blueprint
|
115 |
|
116 |
-
|
117 |
def generate_point_cloud(self, image_name_list):
|
118 |
if not isinstance(image_name_list, list) and not isinstance(image_name_list, str):
|
119 |
raise Exception(
|
|
|
1 |
+
# mast3r.py
|
2 |
import torch
|
3 |
from mini_dust3r.api import OptimizedResult, inferece_dust3r, log_optimized_result
|
4 |
from mini_dust3r.model import AsymmetricCroCo3DStereo
|
|
|
8 |
from mini_dust3r.model import load_model
|
9 |
from pathlib import Path
|
10 |
import uuid
|
11 |
+
from utils import create_image_grid # Corrected import
|
12 |
import rerun as rr
|
13 |
import rerun.blueprint as rrb
|
14 |
import os
|
|
|
113 |
)
|
114 |
return blueprint
|
115 |
|
|
|
116 |
def generate_point_cloud(self, image_name_list):
|
117 |
if not isinstance(image_name_list, list) and not isinstance(image_name_list, str):
|
118 |
raise Exception(
|