聂如
commited on
Commit
·
fedb6c1
1
Parent(s):
69eadb3
Add design file
Browse files
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]
|