8eb4cb520a5ccfca9c25e849ccf0e17b1cd0659d,imgaug/augmenters/geometric.py,PerspectiveTransform,_augment_images,#PerspectiveTransform#Any#Any#Any#Any#,1811

Before Change

            // cv2.warpPerspective only supports <=4 channels
            nb_channels = images[i].shape[2]
            dtype = images[i].dtype
            if dtype not in [np.float32, np.float64, np.uint8]:
                images[i] = images[i].astype(np.float64)  // e.g. np.int32
            if nb_channels <= 4:
                warped = cv2.warpPerspective(images[i], M, (max_width, max_height))
                if warped.ndim == 2 and images[i].ndim == 3:
                    warped = np.expand_dims(warped, 2)

After Change

                h, w = image.shape[0:2]
                warped = ia.imresize_single_image(warped, (h, w), interpolation="cubic")

            if input_dtype == np.bool_:
                warped = warped > 0.5
            elif warped.dtype != input_dtype:
                warped = meta.restore_dtypes_(warped, input_dtype)

            result[i] = warped

        return result

Instances