Update app.py
Browse files
app.py
CHANGED
@@ -249,7 +249,7 @@ def reconstruct_3d_model(images, prompt):
|
|
249 |
multi_view_mask = get_background(normal_multi_view)
|
250 |
rgb_multi_view = rgb_multi_view * rgb_multi_view + (1-multi_view_mask)
|
251 |
input_cameras = get_flux_input_cameras(batch_size=1, radius=4.2, fov=30).to(device_1)
|
252 |
-
vertices, faces = lrm_reconstructions(rgb_multi_view, input_cameras, save_path=save_dir_path, name='lrm', export_texmap=False, if_save_video=
|
253 |
# local normal to global normal
|
254 |
|
255 |
global_normal = local_normal_global_transform(normal_multi_view.permute(0, 2, 3, 1), isomer_azimuths, isomer_elevations)
|
|
|
249 |
multi_view_mask = get_background(normal_multi_view)
|
250 |
rgb_multi_view = rgb_multi_view * rgb_multi_view + (1-multi_view_mask)
|
251 |
input_cameras = get_flux_input_cameras(batch_size=1, radius=4.2, fov=30).to(device_1)
|
252 |
+
vertices, faces = lrm_reconstructions(rgb_multi_view, input_cameras, save_path=save_dir_path, name='lrm', export_texmap=False, if_save_video=True)
|
253 |
# local normal to global normal
|
254 |
|
255 |
global_normal = local_normal_global_transform(normal_multi_view.permute(0, 2, 3, 1), isomer_azimuths, isomer_elevations)
|