f5ff0f5fb3933bb4e704ce7947e2bd5a99ef8358,tests/links_tests/model_tests/yolo_tests/test_yolo_base.py,DummyYOLO,_decode,#DummyYOLO#Any#Any#,34

Before Change

        bboxes = []
        labels = []
        scores = []
        for _ in range(self._value.shape[0]):
            n_bbox = np.random.randint(self._n_anchor - 1)
            bboxes.append(generate_random_bbox(n_bbox, self._insize, 8, 48))
            labels.append(np.random.randint(self._n_fg_class - 1, size=n_bbox)
                          .astype(np.int32))
            scores.append(np.random.uniform(size=n_bbox).astype(np.float32))
        return bboxes, labels, scores


class TestYOLOBase(unittest.TestCase):

After Change

        loc = to_cpu(loc)
        conf = to_cpu(conf)

        if not hasattr(self, "_count"):
            self._count = 0
        np.testing.assert_equal(loc, value[self._count, :, :4])
        np.testing.assert_equal(conf, value[self._count, :, 4:])
        self._count += 1

Instances