【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中详细讲解每个函数模块
      已注释
posted @ 2022-04-22 17:05  freezing001  阅读(310)  评论(0)    收藏  举报