Python horizontal_flipの例

コード例 #1

ファイル: transforms.py プロジェクト: PaddlePaddle/PaddleSeg

    def __call__(self, data):
        if random.random() < self.prob:
            data['img'] = functional.horizontal_flip(data['img'])
            for key in data.get('gt_fields', []):
                data[key] = functional.horizontal_flip(data[key])

        return data

コード例 #2

 def __call__(self, im, im_info=None, label=None):
     if random.random() < self.prob:
         im = functional.horizontal_flip(im)
         if label is not None:
             label = functional.horizontal_flip(label)
     if label is None:
         return (im, im_info)
     else:
         return (im, im_info, label)