tes

inference.py

@@ -31,6 +31,7 @@ def prediction_mask(prediction_result, map_name):
 
     # Get map array 
     map_array = h5_image.get_map(map_name)
+    print("map_array", map_array.shape)
 
     # Convert the map array to a grayscale image
     gray = cv2.cvtColor(map_array, cv2.COLOR_BGR2GRAY)  # greyscale image

test_h5image.py

+from h5Image import H5Image
+
+
+h5_image = H5Image("/projects/bbym/shared/data/commonPatchData/256/OK_250K.hdf5", mode='r', patch_border=0)
+
+map_name = h5_image.get_maps()[0]
+map_array = h5_image.get_map(map_name)
+
+print("map_array.shape", map_array.shape)
\ No newline at end of file