def run_online_evaluation(self, output, target): """ due to deep supervision the return value and the reference are now lists of tensors. We only need the full resolution output because this is what we are interested in in the end. The others are ignored :param output: :param target: :return: """ target = target[0][:, None] # we need to restore color channel dimension here to be compatible with previous code output = output[0] return nnUNetTrainer.run_online_evaluation(self, output, target)
def run_online_evaluation(self, output, target): return nnUNetTrainer.run_online_evaluation(self, output, target)