Skip to content
Snippets Groups Projects
Commit b28707b4 authored by tomrink's avatar tomrink
Browse files

fix bug in scale method

parent 6603fb4a
Branches
No related tags found
No related merge requests found
...@@ -223,7 +223,7 @@ class UNET: ...@@ -223,7 +223,7 @@ class UNET:
f = self.train_label_files f = self.train_label_files
nda = np.load(f) nda = np.load(f)
label = nda[idxs, 0, :, :] label = nda[idxs, label_idx, :, :]
label = np.expand_dims(label, axis=3) label = np.expand_dims(label, axis=3)
else: else:
f = self.test_data_files f = self.test_data_files
... ...
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment