Diffusion扩散模型学习3：Unet学习实现

1、分割常用loss

参考：http://www.bryh.cn/a/186865.html

BCEWithLogitsLoss 二分类，主要就是预测每个像素是否为背景或label

from PIL import Image
# from torch.utils.data.dataset import Dataset
from torchvision import transforms

##加载图片与label mask图片
transform = transforms.Compose([
            transforms.Resize((512, 512)