yanranxiaoxi commited on
Commit
e314b32
·
verified ·
1 Parent(s): cc8681f

fix: planes to image

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -227,16 +227,16 @@ def generate_mesh(image, source_size=512, render_size=384, mesh_size=512, export
227
  return None, None
228
 
229
  def step_1_generate_planes(image):
230
- planes_path, _ = generate_mesh(image)
231
- return planes_path.shape
232
 
233
  def step_2_generate_obj(image):
234
- planes_path, mesh_path = generate_mesh(image, export_mesh=True)
235
- return planes_path.shape, mesh_path, mesh_path
236
 
237
  def step_3_generate_video(image):
238
- planes_path, video_path = generate_mesh(image, export_video=True)
239
- return planes_path.shape, video_path
240
 
241
  # 从 assets 文件夹中设置示例文件,并限制最多读取 10 个文件
242
  example_folder = "assets"
 
227
  return None, None
228
 
229
  def step_1_generate_planes(image):
230
+ planes, _ = generate_mesh(image)
231
+ return planes
232
 
233
  def step_2_generate_obj(image):
234
+ planes, mesh_path = generate_mesh(image, export_mesh=True)
235
+ return planes, mesh_path, mesh_path
236
 
237
  def step_3_generate_video(image):
238
+ planes, video_path = generate_mesh(image, export_video=True)
239
+ return planes, video_path
240
 
241
  # 从 assets 文件夹中设置示例文件,并限制最多读取 10 个文件
242
  example_folder = "assets"