cd219669f1c5296c939bed87a5e211db126811f4,tests/extensions_tests/evaluator_tests/test_detection_coco_evaluator.py,TestDetectionCOCOEvaluatorMPI,setUp,#TestDetectionCOCOEvaluatorMPI#,113

Before Change

        if comm.rank == 0:
            bboxes = [generate_random_bbox(5, (256, 324), 24, 120)
                      for _ in range(10)]
            labels = [2 * np.ones((5,), dtype=np.int32) for _ in range(10)]
        else:
            bboxes = None
            labels = None
        initial_count = comm.rank * batchsize_per_process

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

After Change

        if comm.rank == 0:
            bboxes = [generate_random_bbox(5, (256, 324), 24, 120)
                      for _ in range(10)]
            labels = [np.random.choice(np.arange(3, dtype=np.int32), size=(5,))
                      for _ in range(10)]
        else:
            bboxes = None
            labels = None
        initial_count = comm.rank * batchsize_per_process

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

Instances