JiantaoLin
commited on
Commit
·
476d2e1
1
Parent(s):
4df757e
new
Browse files- app.py +1 -1
- pipeline/utils.py +1 -1
app.py
CHANGED
|
@@ -214,7 +214,7 @@ def image2mesh_main_(reference_3d_bundle_image, caption, seed, strength1=0.5, st
|
|
| 214 |
return gen_save_path, recon_mesh_path
|
| 215 |
# return gen_save_path, recon_mesh_path
|
| 216 |
|
| 217 |
-
@spaces.GPU
|
| 218 |
def bundle_image_to_mesh(
|
| 219 |
gen_3d_bundle_image,
|
| 220 |
lrm_radius = 4.15,
|
|
|
|
| 214 |
return gen_save_path, recon_mesh_path
|
| 215 |
# return gen_save_path, recon_mesh_path
|
| 216 |
|
| 217 |
+
@spaces.GPU(duration=120)
|
| 218 |
def bundle_image_to_mesh(
|
| 219 |
gen_3d_bundle_image,
|
| 220 |
lrm_radius = 4.15,
|
pipeline/utils.py
CHANGED
|
@@ -56,7 +56,7 @@ def lrm_reconstruct(model, infer_config, images,
|
|
| 56 |
else:
|
| 57 |
raise NotImplementedError(f'Unexpected input camera type: {input_camera_type}')
|
| 58 |
|
| 59 |
-
images = v2.functional.resize(images,
|
| 60 |
|
| 61 |
logger.info(f"==> Runing LRM reconstruction ...")
|
| 62 |
planes = model.forward_planes(images, input_cameras)
|
|
|
|
| 56 |
else:
|
| 57 |
raise NotImplementedError(f'Unexpected input camera type: {input_camera_type}')
|
| 58 |
|
| 59 |
+
images = v2.functional.resize(images, 512, interpolation=3, antialias=True).clamp(0, 1)
|
| 60 |
|
| 61 |
logger.info(f"==> Runing LRM reconstruction ...")
|
| 62 |
planes = model.forward_planes(images, input_cameras)
|