2b3db4e19e16e0723130e5dee5d4b668e11a514d,luminoth/models/fasterrcnn/roi_pool_test.py,ROIPoolingTest,testBasic,#ROIPoolingTest#,38

Before Change

        // Construct the pretrained map with four matrix.
        matA = np.ones((5, 5))
        matB = np.ones((5, 5)) + 1
        matC = np.ones((5, 5)) + 2
        matD = np.ones((5, 5)) + 3
        pretrained = np.bmat([[matA, matB], [matC, matD]])
        // Expand the dimensions to be compatible with ROIPoolingLayer.
        pretrained = np.expand_dims(pretrained, axis=0)
        pretrained = np.expand_dims(pretrained, axis=3)

        results = self._run_roi_pooling(roi_proposals, pretrained, self.config)
        print(results["crops"].shape)

After Change

            (4, 2, 2, 1)
        )

        results["roi_pool"] = np.squeeze(results["roi_pool"], axis=3)
        // Check that max polling returns only "multiplier_a"
        self.assertAllEqual(
            results["roi_pool"][0],
            np.ones((2, 2)) * self.multiplier_a

Instances