聂如 commited on
Commit
fedb6c1
·
1 Parent(s): 69eadb3

Add design file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,6 @@ from mast3r.model import AsymmetricMASt3R
14
  import matplotlib.pyplot as pl
15
  from dust3r.utils.image import load_images
16
  import torch.nn.functional as F
17
- from pytorch3d.ops import knn_points
18
  from dust3r.utils.geometry import xy_grid
19
  import numpy as np
20
  import cv2
@@ -187,6 +186,7 @@ def local_get_reconstructed_scene(inputfiles, min_conf_thr, cam_size):
187
  # ])
188
  # os.system('pip install iopath')
189
  # print(f"pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html")
 
190
  outdir = tempfile.mkdtemp(suffix='_FLARE_gradio_demo')
191
  batch = load_images(inputfiles, size=image_size, verbose=not silent)
192
  images = [gt['img'] for gt in batch]
 
14
  import matplotlib.pyplot as pl
15
  from dust3r.utils.image import load_images
16
  import torch.nn.functional as F
 
17
  from dust3r.utils.geometry import xy_grid
18
  import numpy as np
19
  import cv2
 
186
  # ])
187
  # os.system('pip install iopath')
188
  # print(f"pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html")
189
+ from pytorch3d.ops import knn_points
190
  outdir = tempfile.mkdtemp(suffix='_FLARE_gradio_demo')
191
  batch = load_images(inputfiles, size=image_size, verbose=not silent)
192
  images = [gt['img'] for gt in batch]