Ryukijano commited on
Commit
8a533f5
β€’
1 Parent(s): c3c7356

Update mast3r.py

Browse files
Files changed (1) hide show
  1. mast3r.py +2 -3
mast3r.py CHANGED
@@ -1,4 +1,4 @@
1
- # timeforge/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,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 timeforge.utils import create_image_grid
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(