cd219669f1c5296c939bed87a5e211db126811f4,tests/extensions_tests/evaluator_tests/test_instance_segmentation_coco_evaluator.py,TestInstanceSegmentationCOCOEvaluatorMPI,setUp,#TestInstanceSegmentationCOCOEvaluatorMPI#,108

Before Change

        if comm.rank == 0:
            masks = [np.random.uniform(size=(5, 32, 48)) > 0.5
                     for _ in range(10)]
            labels = [np.ones((5,), dtype=np.int32)
                      for _ in range(10)]
        else:
            masks = None
            labels = None
        initial_count = comm.rank * batchsize_per_process

        masks = comm.bcast_obj(masks)
        labels = comm.bcast_obj(labels)
        self.masks = masks
        self.labels = labels

        self.dataset = TupleDataset(
            np.random.uniform(size=(10, 3, 32, 48)),
            masks, labels)
        self.initial_count = initial_count

After Change

        if comm.rank == 0:
            masks = [np.random.uniform(size=(5, 32, 48)) > 0.5
                     for _ in range(10)]
            labels = [np.random.choice(np.arange(3, dtype=np.int32), size=(5,))
                      for _ in range(10)]
        else:
            masks = None
            labels = None
        initial_count = comm.rank * batchsize_per_process

        masks = comm.bcast_obj(masks)
        labels = comm.bcast_obj(labels)
        self.masks = masks
        self.labels = labels

        self.dataset = TupleDataset(
            np.random.uniform(size=(10, 3, 32, 48)),
            masks, labels)
        self.initial_count = initial_count

Instances