语义分割数据增强代码

1.数据增强 

import os
import random
from PIL import Image
import torchvision.transforms as transforms
from tqdm import tqdm
import shutil
import torch
from to_pseudo import process_folder

# 自定义的转换函数
class RandomHorizontalFlipBoth:
    def __init__(self, p=0.5):
        self.transform = transforms.RandomHorizontalFlip(p)

    def __call__(self, img, mask):
        if random.random() < self.transform.p:
            return self.transform(img), self.transform(mask)
        else:
            return img, mask

class RandomVerticalFlipBoth:
    def __init__(self, p=0.5):
        self.transform = transforms.RandomVerticalFlip(p)

    def __call__(self, img, mask):
        if random.random() < self.transform.p:
            return self.