DUMP_IMG

handcrafted_solution.py

@@ -6,13 +6,15 @@ import numpy as np
 from collections import defaultdict
 import cv2
 from typing import Tuple, List
-from scipy.sparse import random
+
 from scipy.spatial.distance import cdist
 
 from hoho.read_write_colmap import read_cameras_binary, read_images_binary, read_points3D_binary
 from hoho.color_mappings import gestalt_color_mapping, ade20k_color_mapping
 
 DUMP_IMG = False
+if DUMP_IMG:
+  from scipy.sparse import random
 
 def empty_solution():
     '''Return a minimal valid solution, i.e. 2 vertices and 1 edge.'''
@@ -234,9 +236,9 @@ def get_vertices_and_edges_from_two_segmentations(ade_seg_np, gest_seg_np, edge_
                 apex_map_on_gest[uu+ss[0], vv+ss[1]] = (255,0,0)
 
     # imsave apex and eave_end
-    import random
-    rid = random.random()
     if DUMP_IMG:
+      import random
+      rid = random.random()
       filename_apex_ade = f'apex_map_on_ade_{rid}.jpg'
       cv2.imwrite(filename_apex_ade, apex_map_on_ade)
       filename_apex_gest = f'apex_map_on_gest_{rid}.jpg'
@@ -244,7 +246,7 @@ def get_vertices_and_edges_from_two_segmentations(ade_seg_np, gest_seg_np, edge_
       filename_apex_map = f'apex_map_{rid}.jpg'
       cv2.imwrite(filename_apex_map, apex_map)
 
-    print(f'{len(vertices)} vertices detected')
+    #print(f'{len(vertices)} vertices detected')
     # Connectivity
     apex_pts = []
     apex_pts_idxs = []
@@ -447,7 +449,7 @@ def predict(entry, visualize=False) -> Tuple[np.ndarray, List[int]]:
         vertices_3d = cv2.convertPointsFromHomogeneous(vertices_3d).reshape(-1, 3)
         vert_edge_per_image[i] = vertices, connections, vertices_3d
     all_3d_vertices, connections_3d = merge_vertices_3d(vert_edge_per_image, 3.0)
-    print(f'after merge, {len(all_3d_vertices)} 3d vertices and {len(connections_3d)} 3d connections')
+    #print(f'after merge, {len(all_3d_vertices)} 3d vertices and {len(connections_3d)} 3d connections')
     #all_3d_vertices_clean, connections_3d_clean  = prune_not_connected(all_3d_vertices, connections_3d)
     all_3d_vertices_clean, connections_3d_clean  = all_3d_vertices, connections_3d # don't prune -> cost:2.0
     #print(f'after pruning, {len(all_3d_vertices_clean)} 3d clean vertices and {len(connections_3d_clean)} 3d clean connections')