d381fb0787395076ecb8a3bd6984f52a16793fae,dataset/batch_image.py,ImagesBatch,_cutout_,#ImagesBatch#Any#Any#Any#Any#,822

Before Change

        image_shape = self._get_image_shape(image)
        shape = _get_shape(shape)
        origin = _get_origin(shape, origin)
        right_bottom = (min(origin[0] + shape[0], image_shape[0]),
                        min(origin[1] + shape[1], image_shape[1]))
        image[origin[0]:right_bottom[0], origin[1]:right_bottom[1]] = color

        return image

After Change

        image = image.copy()
        shape = (shape, shape) if isinstance(shape, Number) else shape
        origin = self._calc_origin(shape, origin, self._get_image_shape(image))
        image.paste(PIL.Image.new("RGB", shape, color), tuple(origin))
        return image
        // def _get_shape(shape):
        //     return (shape, shape) if isinstance(shape, Number) else shape

Instances