ef33967e472776ef4c4b20bb6c5e6961881bd5fc,chainercv/extensions/evaluator/semantic_segmentation_evaluator.py,SemanticSegmentationEvaluator,evaluate,#SemanticSegmentationEvaluator#,72

Before Change

    def evaluate(self):
        iterator = self._iterators["main"]
        target = self._targets["main"]

        if hasattr(iterator, "reset"):
            iterator.reset()
            it = iterator
        else:
            it = copy.copy(iterator)

        in_values, out_values, rest_values = apply_to_iterator(
            target.predict, it)
        // delete unused iterators explicitly
        del in_values

        pred_labels, = out_values

After Change

        self.comm = comm

    def evaluate(self):
        target = self._targets["main"]
        if self.comm is not None and self.comm.rank != 0:
            apply_to_iterator(target.predict, None, comm=self.comm)
            return {}
        iterator = self._iterators["main"]

        if hasattr(iterator, "reset"):
            iterator.reset()
            it = iterator
        else:
            it = copy.copy(iterator)

        in_values, out_values, rest_values = apply_to_iterator(
            target.predict, it, comm=self.comm)
        // delete unused iterators explicitly
        del in_values

        pred_labels, = out_values

Instances