Update app.py

app.py

@@ -59,7 +59,7 @@ def generate_point_cloud(color_img):
     depth_img = predict_depth(color_img[:, :, ::-1])
     # 画像サイズの調整
     height, width = color_img.shape[:2]
-    new_height = 1600
+    new_height = 512
     new_width = int(width * (new_height / height))
     
     color_img_resized = np.array(Image.fromarray(color_img).resize((new_width, new_height), Image.LANCZOS))
@@ -74,7 +74,7 @@ def generate_point_cloud(color_img):
     adjusted_depth = np.power(normalized_depth, 0.1)  # ガンマ補正
     
     # カメラの内部パラメータ（使用するカメラに基づいて調整）
-    fx, fy = 1000, 1000  # 焦点距離
+    fx, fy = 300, 300  # 焦点距離
     cx, cy = color_img_resized.shape[1] / 2, color_img_resized.shape[0] / 2  # 主点
 
     # メッシュグリッドの作成
@@ -101,6 +101,9 @@ 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)