医学图像分割---UNet（Pytorch）2021-06-19

最新推荐文章于 2024-09-25 20:16:15 发布

奋斗的张

最新推荐文章于 2024-09-25 20:16:15 发布

阅读量847

点赞数

分类专栏：医学图像分割文章标签：深度学习 pytorch 神经网络机器学习

本文链接：https://blog.csdn.net/weixin_47035018/article/details/118048990

版权

本文介绍了如何利用Pytorch实现UNet模型进行医学图像分割，提供了详细的github链接供读者查阅和训练网络。对于关注医学图像分割和深度学习的学者，这是一个有价值的交流资源。

摘要由CSDN通过智能技术生成

医学图像分割—UNet（Pytorch）

github链接（欢迎多多star）修改文件路径就可以直接训练网络

有医学图像分割，图像分割研究兴趣的同学，欢迎交流！

import torch
import torch.nn as nn
import numpy as np
import torch.nn.functional as F


class DoubleConv(nn.Module):
    def __init__(self, in_ch, out_ch):
        super(DoubleConv, self).__init__()
        self.conv = nn.Sequential(
            nn.Conv2d(in_ch, out_ch, 3, padding=1),
            nn.BatchNorm2d(out_ch),
            nn.ReLU(inplace=True),
            nn.Conv2d(out_ch, out_ch, 3, padding=1),
            nn.BatchNorm2d(out_ch),
            nn.ReLU(inplace=True)
        )

    def forward(self, input):
        return self.conv(input)


class