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

2024-01-09 18:58:57

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)

文章来源:https://blog.csdn.net/weixin_42357472/article/details/135449452
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。