8bdee2eb948e7c41a2df4532ea009c661d6bd3ac,imgaug/augmenters/arithmetic.py,AddElementwise,_augment_images,#AddElementwise#Any#Any#Any#Any#,227

Before Change

                samples = self.value.draw_samples((height, width, 1), random_state=rs_image)
                samples = np.tile(samples, (1, 1, nb_channels))
            after_add = image + samples
            result[i] = after_add

        // TODO make value range more flexible
        meta.clip_augmented_images_(result, 0, 255)
        meta.restore_augmented_images_dtypes_(result, input_dtypes)

        return result

    def _augment_keypoints(self, keypoints_on_images, random_state, parents, hooks):

After Change

                samples = np.tile(samples, (1, 1, nb_channels))
            after_add = image + samples

            after_add = meta.clip_augmented_image_(after_add, 0, 255) // TODO make value range more flexible
            after_add = meta.restore_augmented_image_dtype_(after_add, input_dtypes[i])

            result[i] = after_add

        return result

Instances