Skip to content
Snippets Groups Projects
Commit 3ca30ea2 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Merge branch 'preprocessor_bug_fix' into 'master'

Adapted to new h5py interface

See merge request !62
parents 194c4789 cb6826f8
Branches
Tags
1 merge request!62Adapted to new h5py interface
Pipeline #61539 failed
...@@ -82,12 +82,11 @@ class Preprocessor(BasePreprocessor): ...@@ -82,12 +82,11 @@ class Preprocessor(BasePreprocessor):
def write_data(self, data, filename): def write_data(self, data, filename):
"""Overrides the default method implementation to handle our tuple""" """Overrides the default method implementation to handle our tuple"""
f = h5py.File(filename, "w") with h5py.File(filename, "w") as f:
f.set("image", data[0]) f["image"] = data[0]
f.set("mask", data[1]) f["mask"] = data[1]
def read_data(self, filename): def read_data(self, filename):
"""Overrides the default method implementation to handle our tuple""" """Overrides the default method implementation to handle our tuple"""
with h5py.File(filename, "r") as f:
f = h5py.File(filename, "r") return f["image"][()], f["mask"][()]
return f.read("image"), f.read("mask")
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment