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