def __call__()

in siammot/data/adapters/augmentation/video_augmentation.py [0:0]

• 14 lines of code
• 7 McCabe index (conditional complexity)

    def __call__(self, video, target):
        # Blur augmentation only applies for Siamese Training
        if not isinstance(video, (list, tuple)) or self.motion_blur_prob == 0.0:
            return video, target

        new_video = []
        new_target = []
        idx = random.choice((0, 1))
        for i, (image, image_target) in enumerate(zip(video, target)):
            if i == idx:
                random_prob = random.uniform(0, 1)
                if random_prob < self.motion_blur_prob:
                    image = self.motion_blur_func(image)
            new_video.append(image)
            new_target.append(image_target)

        return new_video, new_target