NifTK / NiftyNet

[unmaintained] An open-source convolutional neural networks platform for research in medical image analysis and image-guided therapy
http://niftynet.io
Apache License 2.0
1.36k stars 403 forks source link

403 debug multioutput multibatch nnhack #404

Closed csudre closed 5 years ago

csudre commented 5 years ago

Status

READY

Description

Extends the aggregators to support multiple outputs (images and CSVs) at each decoding iteration.

This PR will affect existing applications' interpret_output method. Previous applications call

self.output_decoder.decode_batch(
    batch_output['window'], 
    batch_output['location'])

should be replaced by a dictionary parameter:

self.output_decoder.decode_batch(
    {'window_seg': batch_output['window']}, 
    batch_output['location'])

If the key of the dictionary contains"window", the corresponding value will be saved as images (.nii.gz). The value will be saved as csv files otherwise.

Types of changes

Impacted Areas in Application

List general components of the application that this PR will affect: