摘要: 和FCN类似,UNet是另一个做语义分割的网络,网络从输入到输出中间呈一个U型而得名。 相比于FCN,UNet增加了更多的中间连接,能够更好处理不同尺度上的特征。 网络结构如下: 下面代码是用UNet对VOC数据集做的语义分割。 import torch import torch.nn as nn 阅读全文
posted @ 2024-10-01 15:10 Dsp Tian 阅读(140) 评论(0) 推荐(1) 编辑
摘要: 深度学习中有些数据集可能不符合imagenet计算出的均值和标准差,需要根据自己的数据集单独计算。 下面这个脚本能够计算当前数据集均值和标准差。 import torch import os from PIL import Image from torchvision import transfor 阅读全文
posted @ 2024-10-01 13:21 Dsp Tian 阅读(22) 评论(0) 推荐(0) 编辑