Update app.py

app.py

@@ -55,11 +55,11 @@ import numpy as np
 import trimesh
 import tempfile
 
-def generate_point_cloud(color_img):
+def generate_point_cloud(color_img, resolution):
     depth_img = predict_depth(color_img[:, :, ::-1])
     # 画像サイズの調整
     height, width = color_img.shape[:2]
-    new_height = 512
+    new_height = resolution
     new_width = int(width * (new_height / height))
     
     color_img_resized = np.array(Image.fromarray(color_img).resize((new_width, new_height), Image.LANCZOS))
@@ -73,7 +73,7 @@ def generate_point_cloud(color_img):
     # 非線形変換（必要に応じて調整）
     adjusted_depth = np.power(normalized_depth, 0.1)  # ガンマ補正
     
-    # カメラの内部パラメータ（使用するカメラに基づいて調整）
+    # カメラの内部パラメータ（使用するカメラに基づいて調整）    
     fx, fy = 300, 300  # 焦点距離
     cx, cy = color_img_resized.shape[1] / 2, color_img_resized.shape[0] / 2  # 主点
 
@@ -101,9 +101,6 @@ def generate_point_cloud(color_img):
     # PointCloudオブジェクトの作成
     cloud = trimesh.PointCloud(vertices=points, colors=colors)
 
-    # PointCloudオブジェクトの作成
-    cloud = trimesh.PointCloud(vertices=points, colors=colors)
-
     # Z軸周りに180度回転を適用（時計回り）
     rotation = trimesh.transformations.rotation_matrix(np.pi, [0, 0, 1])
     cloud.apply_transform(rotation)
@@ -125,7 +122,8 @@ with gr.Blocks(css=css) as demo:
 
     with gr.Row():
         input_image = gr.Image(label="Input Image", type='numpy', elem_id='img-display-input')
-    
+        resolution_slider = gr.Slider(minimum=512, maximum=1600, value=512, step=1, label="Resolution")
+
     submit = gr.Button(value="Compute Depth & Generate Point Cloud")
     
     output_3d = gr.Model3D(
@@ -133,7 +131,7 @@ with gr.Blocks(css=css) as demo:
         label="3D Model",
     )
     
-    submit.click(fn=generate_point_cloud, inputs=[input_image], outputs=[output_3d])
+    submit.click(fn=generate_point_cloud, inputs=[input_image, resolution_slider], outputs=[output_3d])
 
 if __name__ == '__main__':
     demo.queue().launch(share=True)