5a29571bdeb6deaffa8ed1e7b404566bc7b62505,niftynet/io/image_reader.py,ImageReader,initialise_reader,#ImageReader#Any#Any#,40

Before Change

        
        app_type = task_param.name

        if app_type == "net_segment.py":
            from niftynet.application.segmentation_application \
                import SUPPORTED_INPUT
            // only choose output names that are supported by the application
            // (SUPPORTED_INPUT) and have user parameter specification
        elif app_type == "net_gan.py":
            from niftynet.application.gan_application \
                import SUPPORTED_INPUT

        if not self.names:
            // by default, reader tries to output all supported output names
            self.names = SUPPORTED_INPUT

After Change

        data_to_load = {}
        for name in self._names:
            for source in self._input_sources[name]:
                if source not in data_param:
                    tf.logging.fatal(
                        "reader name [{}] requires [{}], however it is not "
                        "specified as a section in the config, "
                        "current input section names: {}".format(
                            name, source, list(data_param)))
                    raise ValueError
                else:
                    data_to_load[source] = data_param[source]
        self._file_list = ImageReader.load_and_merge_csv_files(data_to_load)
        self.output_list = filename_to_image_list(self._file_list,
                                                  self._input_sources,
                                                  data_param)

Instances