f22c4ebf74fd00bb8e2ec136ea9cf0f7177ebf61,TestImageAugmenter.py,TestImageAugmenter,test_two_channels,#TestImageAugmenter#,186

Before Change

        
        images = np.array([image_before]).astype(np.uint8)
        nb_similar = 0
        for _ in range(100):
            image_after = ia.augment_batch(images)[0]
            //print(image_after)
            if np.allclose(image_target, image_after):
                nb_similar += 1
        //print("similar:",nb_similar)
        self.assertTrue(nb_similar > (33-10) and nb_similar < (33+10))
        
        
        // two channels, channel is last axis of each image

After Change

        images = np.array([image_before]).astype(np.uint8)
        
        nb_augment = 1000
        images_augmented = ia.augment_batch(np.resize(images, (nb_augment, 2, 2, 2)))
        
        nb_similar = 0
        for image_after in images_augmented:

Instances