首页
搜索 搜索
当前位置:资讯中心 > 正文

当前快看:萌新的卷积神经网络学习笔记--搭建Unet

2023-01-26 23:47:09 哔哩哔哩

导入需要的模块


【资料图】

pytorch是比较常用的深度学习框架,集成类似numpy的功能,生成以及操作叫做张量的多维数组。张量(tensor):能放在GPU上跑的多维数组(大概)

搭建网络

网络基础框架

全连接:图片数据需要转化成向量输入,如[1,2,3,4...m],size:1*m,之后一般经过一个非线性函数激活即数值归一化到0~1或-1~1之间

卷积:随机初始化一个卷积核(卷积核有提取特征的作用),铺在图片上做滑窗操作,对应的像素相乘再求和,之后一般经过一个非线性函数激活即数值归一化到0~1或-1~1之间(大概)

池化:对n*n区域内像素进行操作,求最大值或求平均,得到一个数值代表该区域(大概)

Unet

查看是否能跑通

输出:

不知道对不对,反正能跑通!