Lets try to not import many packages.

predict.py

@@ -3,17 +3,17 @@ from typing import Tuple, List
 from hoho2025.example_solutions import empty_solution, read_colmap_rec, get_vertices_and_edges_from_segmentation, get_house_mask, fit_scale_robust_median, get_uv_depth, merge_vertices_3d, prune_not_connected, prune_too_far, point_to_segment_dist
 from hoho2025.color_mappings import ade20k_color_mapping, gestalt_color_mapping
 from PIL import Image, ImageDraw
-from visu import save_gestalt_with_proj, draw_crosses_on_image
+#from visu import save_gestalt_with_proj, draw_crosses_on_image
 import os
 import pycolmap
 from PIL import Image as PImage
 import cv2
-import open3d as o3d
-from visu import plot_reconstruction_local, plot_wireframe_local, plot_bpo_cameras_from_entry_local
-import pyvista as pv
+#import open3d as o3d
+#from visu import plot_reconstruction_local, plot_wireframe_local, plot_bpo_cameras_from_entry_local
+#import pyvista as pv
 from fast_pointnet import save_patches_dataset, predict_vertex_from_patch
-from fast_voxel import predict_vertex_from_patch_voxel
-import time
+#from fast_voxel import predict_vertex_from_patch_voxel
+#import time
 from fast_pointnet_class import save_patches_dataset as save_patches_dataset_class
 from fast_pointnet_class import predict_class_from_patch
 from scipy.spatial.distance import cdist