NestedUNet_Pytorch.zip

基于Pytorch的UNet、UNet++的代码.参考csdn的链接:https://blog.csdn.net/weixin_40519315/article/details/104581992

# -*- coding: utf-8 -*-
import numpy as np

from torch import nn
from torch.nn import functional as F
import torch
from torchvision import models
import torchvision


class VGGBlock（nn.Module）:
    def __init__（self in_channels middle_channels out_channels act_func=nn.ReLU（inplace=True））:
        super（VGGBlock self）.__init__（）
        self.act_func = act_func
        self.conv1 = nn.Conv2d（in_channels middle_channels 3 padding=1）
        self.bn1 = nn.BatchNorm2d（middle_channels）
        self.conv2 = nn.Conv2d（middle_channels out_channels 3 padding=1）
        self.bn2 = nn.BatchNorm2d（out_channels）

    def forward（self x）:
        out = self.conv1（x）
        out = self.bn1（out）
        out = self.act_func（out）

        out = self.conv2（out）
        out = self.bn2（out）
       

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2020-03-04 16:26  NestedUNet_Pytorch\
     文件        5710  2019-03-11 21:11  NestedUNet_Pytorch\archs.py
     文件        1309  2020-03-04 16:10  NestedUNet_Pytorch\dataset.py
     目录           0  2020-03-04 16:25  NestedUNet_Pytorch\input\
     文件           0  2019-03-11 21:11  NestedUNet_Pytorch\input\.gitkeep
     文件        1047  2019-03-11 21:11  NestedUNet_Pytorch\losses.py
     文件        3999  2019-03-11 21:11  NestedUNet_Pytorch\metrics.py
     目录           0  2020-03-04 16:25  NestedUNet_Pytorch\models\
     文件           0  2019-03-11 21:11  NestedUNet_Pytorch\models\.gitkeep
     目录           0  2020-03-04 16:25  NestedUNet_Pytorch\output\
     文件           0  2019-03-11 21:11  NestedUNet_Pytorch\output\.gitkeep
     文件        1536  2019-03-11 21:11  NestedUNet_Pytorch\preprocess_dsb2018.py
     文件          66  2020-03-04 16:26  NestedUNet_Pytorch\README.md
     文件        3585  2020-03-04 14:38  NestedUNet_Pytorch\test.py
     文件        9591  2020-03-04 14:45  NestedUNet_Pytorch\train.py
     文件         306  2019-03-11 21:11  NestedUNet_Pytorch\utils.py