389970fff42bbf094dcc494b6b7f00e1ebd6f800,tests/extensions_tests/detection_tests/test_detection_vis_report.py,TestDetectionVisReport,setUp,#TestDetectionVisReport#,23

Before Change

        self.trainer.out = self.out_dir
        self.trainer.updater.iteration = 0

        model = ConstantReturnModel(np.random.uniform(size=(1,) + self.shape))
        dataset = DummyDataset(
            shapes=[(3, 10, 10), self.shape],
            dtypes=[np.float32, np.float32])

        self.extension = DetectionVisReport(
            self.indices, dataset, model,
            filename_base="detection")

After Change

        model = ConstantReturnModel(
            (np.random.uniform(size=(1,) + self.bbox_shape),
             np.random.uniform(size=(1,) + self.label_shape)))
        dataset = DummyDataset(
            shapes=[(3, 10, 10), self.bbox_shape, self.label_shape],
            dtypes=[np.float32, np.float32, np.int32])

        self.extension = DetectionVisReport(
            self.indices, dataset, model,
            filename_base="detection")

Instances