语义分割时如何对原图和label做相同的裁剪

import os
import torch
import numpy as np
from torch.autograd import Variable
from torch import nn
import torch.nn.functional as F
from torch.utils.data import Dataset, DataLoader
from PIL import Image，ImageFilter,ImageDraw
from torchvision import transforms as tfs
from datetime import datetime
import matplotlib.pyplot as plt
from tqdm import tqdm
import random

#crop size
img_w = 256  
img_h = 256
#crop function
def rand_crop(data,label):
   
    width1 = random.randint(0, data.size[0] - img_w )
    height1 = random.randint(0, data.size[1] - img_h)
    width2 = width1 + img_w
    height2 = height1 + img_h 
            
    data=data.crop((width1, height1, width2, height2))
    label=label.crop((width1, height1, width2, height2))

    return data,labe

原文：https://blog.youkuaiyun.com/qq_41997920/article/details/89333190