Spaces:
Runtime error
Runtime error
kxhit
commited on
Commit
·
8db7bce
1
Parent(s):
bb54316
rembg version
Browse files- app.py +5 -3
- dust3r/utils/image.py +2 -0
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -562,7 +562,7 @@ Image views are treated as tokens and the camera pose is encoded by <b>CaPE (Cam
|
|
| 562 |
|
| 563 |
<h4><b>Tips:</b></h4>
|
| 564 |
|
| 565 |
-
- Our model can take <b>any number input images</b>. The more images you provide (>=3 for this demo)
|
| 566 |
|
| 567 |
- Our model can generate <b>any number and any pose</b> novel views. You can specify the number of views you want to generate. In this demo, we set novel views on an <b>archemedian spiral</b> for simplicity.
|
| 568 |
|
|
@@ -619,7 +619,7 @@ with gr.Blocks() as demo:
|
|
| 619 |
|
| 620 |
with gr.Row():
|
| 621 |
gr.Markdown('''
|
| 622 |
-
<h4><b>Check if the pose and segmentation looks correct. If not, remove the incorrect images and try again.</b></h4>
|
| 623 |
''')
|
| 624 |
|
| 625 |
with gr.Row():
|
|
@@ -667,7 +667,9 @@ with gr.Blocks() as demo:
|
|
| 667 |
)
|
| 668 |
|
| 669 |
with gr.Row():
|
| 670 |
-
gr.Markdown('''
|
|
|
|
|
|
|
| 671 |
|
| 672 |
gr.Markdown(_CITE_)
|
| 673 |
|
|
|
|
| 562 |
|
| 563 |
<h4><b>Tips:</b></h4>
|
| 564 |
|
| 565 |
+
- Our model can take <b>any number input images</b>. The more images you provide <b>(>=3 for this demo)</b>, the better the results.
|
| 566 |
|
| 567 |
- Our model can generate <b>any number and any pose</b> novel views. You can specify the number of views you want to generate. In this demo, we set novel views on an <b>archemedian spiral</b> for simplicity.
|
| 568 |
|
|
|
|
| 619 |
|
| 620 |
with gr.Row():
|
| 621 |
gr.Markdown('''
|
| 622 |
+
<h4><b>Check if the pose (blue is axis is estimated z-up direction) and segmentation looks correct. If not, remove the incorrect images and try again.</b></h4>
|
| 623 |
''')
|
| 624 |
|
| 625 |
with gr.Row():
|
|
|
|
| 667 |
)
|
| 668 |
|
| 669 |
with gr.Row():
|
| 670 |
+
gr.Markdown('''
|
| 671 |
+
<h4><b>The novel views are generated on an archimedean spiral (rotating around z-up axis and looking at the object center). You can download the video.</b></h4>
|
| 672 |
+
''')
|
| 673 |
|
| 674 |
gr.Markdown(_CITE_)
|
| 675 |
|
dust3r/utils/image.py
CHANGED
|
@@ -111,6 +111,7 @@ def load_images(folder_or_list, size, square_ok=False, verbose=True, do_remove_b
|
|
| 111 |
|
| 112 |
imgs = []
|
| 113 |
imgs_rgba = []
|
|
|
|
| 114 |
for path in folder_content:
|
| 115 |
if not path.lower().endswith(supported_images_extensions):
|
| 116 |
continue
|
|
@@ -169,4 +170,5 @@ def load_images(folder_or_list, size, square_ok=False, verbose=True, do_remove_b
|
|
| 169 |
assert imgs, 'no images foud at '+root
|
| 170 |
if verbose:
|
| 171 |
print(f' (Found {len(imgs)} images)')
|
|
|
|
| 172 |
return imgs, imgs_rgba
|
|
|
|
| 111 |
|
| 112 |
imgs = []
|
| 113 |
imgs_rgba = []
|
| 114 |
+
load_time = time.time()
|
| 115 |
for path in folder_content:
|
| 116 |
if not path.lower().endswith(supported_images_extensions):
|
| 117 |
continue
|
|
|
|
| 170 |
assert imgs, 'no images foud at '+root
|
| 171 |
if verbose:
|
| 172 |
print(f' (Found {len(imgs)} images)')
|
| 173 |
+
print(f' (Loading time: {time.time()-load_time:.2f}s)')
|
| 174 |
return imgs, imgs_rgba
|
requirements.txt
CHANGED
|
@@ -6,7 +6,7 @@ accelerate
|
|
| 6 |
transformers
|
| 7 |
gradio
|
| 8 |
spaces
|
| 9 |
-
rembg==2.0.
|
| 10 |
open3d
|
| 11 |
trimesh
|
| 12 |
einops
|
|
|
|
| 6 |
transformers
|
| 7 |
gradio
|
| 8 |
spaces
|
| 9 |
+
rembg==2.0.56
|
| 10 |
open3d
|
| 11 |
trimesh
|
| 12 |
einops
|