fixes segmentation app style
@@ -32,8 +32,8 @@ from niftynet.layer.rgb_histogram_equilisation import \
from niftynet.evaluation.segmentation_evaluator import SegmentationEvaluator
from niftynet.layer.rand_elastic_deform import RandomElasticDeformationLayer
-SUPPORTED_INPUT = set(['image', 'label', 'weight', 'sampler', 'inferred',
- 'value'])
+SUPPORTED_INPUT = set(
+ ['image', 'label', 'weight', 'sampler', 'inferred', 'value'])
class SegmentationApplication(BaseApplication):
@@ -482,8 +482,8 @@ class SegmentationApplication(BaseApplication):
def interpret_output(self, batch_output):
if self.is_inference:
return self.output_decoder.decode_batch(
- {'window_seg': batch_output['window']}, batch_output[
- 'location'])
+ {'window_seg': batch_output['window']},
+ batch_output['location'])
return True
