【B站学习(存档)】使用Pytorch搭建U-Net网络并基于DRIVE数据集训练(语义分割)-1网络的搭建
B站链接:https://www.bilibili.com/video/BV1rq4y1w7xM?spm_id_from=333.880.my_history.page.click
-
比较主流的方式是经过卷积层之后不改变高和宽。
并将转置卷积替换成双线性插值
![]()
-
数据情况:mask提供的是二值图片,提供感兴趣的区域,相当于一个蒙版的作用。
-
注意修改数据的路径
-
【讲解网络搭建的部分】
- 首先在src目录下的unet.py中详细讲解每个函数模块
已注释
- 首先在src目录下的unet.py中详细讲解每个函数模块


浙公网安备 33010602011771号