Esempio n. 1
0
    def __init__(self, p: float = 0.5):
        """Initialization

        Args:
            p (float, optional): Probability to apply this transformation. Defaults to 0.5.
        """
        self.flip_coin_vflip = ops.CoinFlip(probability=p)
        self.image_vflip = ops.Flip(device='gpu', horizontal=0)
        self.bbox_vflip = ops.BbFlip(device='cpu', horizontal=0)
        self.ldmrks_vflip = ops.CoordFlip(layout='xy', device='cpu', flip_x=0)
Esempio n. 2
0
 def __init__(self, device, batch_size, iterator, layout,
              center_x = None, center_y = None, center_z = None,
              num_threads=1, device_id=0):
     super(CoordFlipPipeline, self).__init__(batch_size, num_threads, device_id)
     self.device = device
     self.iterator = iterator
     self.coord_flip = ops.CoordFlip(device = self.device, layout=layout,
                                     center_x=center_x, center_y=center_y, center_z=center_z)
     self.flip_x = ops.CoinFlip(probability = 0.5)
     self.flip_y = ops.CoinFlip(probability = 0.5)
     self.flip_z = ops.CoinFlip(probability = 0.5) if len(layout) == 3 else None