kxhit commited on
Commit
8db7bce
1 Parent(s): bb54316

rembg version

Browse files
Files changed (3) hide show
  1. app.py +5 -3
  2. dust3r/utils/image.py +2 -0
  3. 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), 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,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('''The novel views are generated on an archimedean spiral. You can download the video''')
 
 
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.50
10
  open3d
11
  trimesh
12
  einops
 
6
  transformers
7
  gradio
8
  spaces
9
+ rembg==2.0.56
10
  open3d
11
  trimesh
12
  einops