Spaces:
Running
on
Zero
Running
on
Zero
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
|