Skip to content
Snippets Groups Projects
Commit 08e5664b authored by Nattapon Jaroenchai's avatar Nattapon Jaroenchai
Browse files

Update test_h5image.py

parent 1c83e396
No related branches found
No related tags found
No related merge requests found
......@@ -21,7 +21,7 @@ if dset.ndim == 3:
image = dset[...].transpose(2, 0, 1) # rasterio expects bands first
else:
image = np.array(dset[...], ndmin=3)
rasterios.open(f"{map_name}.tif", 'w', driver='GTiff', compress='lzw',
rasterios.open(f"/projects/bbym/nathanj/attentionUnet/infer_results/{map_name}.tif", 'w', driver='GTiff', compress='lzw',
height=image.shape[1], width=image.shape[2], count=image.shape[0], dtype=image.dtype,
crs=h5_image.get_crs(map_name, legend), transform=h5_image.get_transform(map_name, legend)).write(image)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment