Add map size

inference.py

@@ -252,6 +252,7 @@ def main(args):
     
     # Get the size of the map
     map_width, map_height, _ = h5_image.get_map_size(map_name)
+    print("Map size:", h5_image.get_map_size(map_name))
     
     # Calculate the number of patches based on the patch size and border
     num_rows = math.ceil(map_width / h5_image.patch_size)