Commit e6ffde5a authored by mibaumgartner's avatar mibaumgartner
Browse files

remove seg cast when unpacking

parent 1235f737
...@@ -206,7 +206,7 @@ def npz2npy(npz_file: str, delete_npz: bool = False): ...@@ -206,7 +206,7 @@ def npz2npy(npz_file: str, delete_npz: bool = False):
a = load_npz_looped(npz_file, keys=["data", "seg"], num_tries=3) a = load_npz_looped(npz_file, keys=["data", "seg"], num_tries=3)
if a is not None: if a is not None:
np.save(npz_file[:-3] + "npy", a["data"]) np.save(npz_file[:-3] + "npy", a["data"])
np.save(npz_file[:-4] + "_seg.npy", a["seg"].astype(np.int16)) np.save(npz_file[:-4] + "_seg.npy", a["seg"])
if delete_npz: if delete_npz:
os.remove(npz_file) os.remove(npz_file)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment