请选择 进入手机版 | 继续访问电脑版

Hello Mat

 找回密码
 立即注册

QQ登录

只需一步,快速开始

12
返回列表 发新帖
楼主: GTX_AI

Unet网络结构

[复制链接]

15

主题

38

帖子

947

积分

管理员

Rank: 9Rank: 9Rank: 9

积分
947
 楼主| 发表于 2019-6-23 22:21:30 | 显示全部楼层
small 网络1::模型大小:65474KB,63.9MB
  1. import torch.nn as nn
  2. import torch
  3. from torch import autograd
  4. import torchvision.models as models


  5. # 导入模型结构
  6. #resnet50 = models.resnet50(pretrained=True)
  7. # 加载预先下载好的预训练参数到resnet18
  8. #resnet50.load_state_dict(torch.load('resnet50-5c106cde.pth'))

  9. class DoubleConv(nn.Module):
  10.     def __init__(self, in_ch, out_ch):
  11.         super(DoubleConv, self).__init__()
  12.         self.conv = nn.Sequential(
  13.             nn.Conv2d(in_ch, out_ch, 3, padding=1),
  14.             nn.BatchNorm2d(out_ch),
  15.             nn.ReLU(inplace=True),
  16.             nn.Conv2d(out_ch, out_ch, 3, padding=1),
  17.             nn.BatchNorm2d(out_ch),
  18.             nn.ReLU(inplace=True)
  19.         )

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


  22. class Unet(nn.Module):
  23.     def __init__(self,in_ch,out_ch):
  24.         super(Unet, self).__init__()
  25.         # 256x256
  26.         self.conv1 = DoubleConv(in_ch, 192)
  27.         self.pool1 = nn.MaxPool2d(2)
  28.         self.conv2 = DoubleConv(192, 384)
  29.         self.pool2 = nn.MaxPool2d(2)
  30.         self.conv3 = DoubleConv(384, 768)
  31.         self.up4 = nn.ConvTranspose2d(768, 384, 2, stride=2)
  32.         self.conv4 = DoubleConv(768, 384)
  33.         self.up5 = nn.ConvTranspose2d(384, 192, 2, stride=2)
  34.         self.conv5 = DoubleConv(384, 192)
  35.         self.conv6 = nn.Conv2d(192,out_ch, 1)

  36.     def forward(self,x):
  37.         c1=self.conv1(x)
  38.         p1=self.pool1(c1)
  39.         c2=self.conv2(p1)
  40.         p2=self.pool2(c2)
  41.         c3=self.conv3(p2)
  42.         up_4=self.up4(c3)
  43.         merge4 = torch.cat([up_4, c2], dim=1)
  44.         c4=self.conv4(merge4)
  45.         up_5=self.up5(c4)
  46.         merge5 = torch.cat([up_5, c1],dim=1)
  47.         c5=self.conv5(merge5)
  48.         c6=self.conv6(c5)
  49.         out = nn.Sigmoid()(c6)
  50.         return out
复制代码


回复 支持 反对

使用道具 举报

15

主题

38

帖子

947

积分

管理员

Rank: 9Rank: 9Rank: 9

积分
947
 楼主| 发表于 2019-6-25 22:17:52 | 显示全部楼层
总结:
extra_large的Unet网络结构:模型大小:1944756KB,1.85GB,Size:16~4096
large Unet网络1:模型大小:121652KB,                  118MB,Size:16~1024
large Unet网络2::模型大小:486257KB,               474MB,Size:32~2048
large Unet网络::模型大小:121329KB,                 118MB,Size:64~1024
normal网络::模型大小:30126KB,                        29.4MB,Size:64~512
normal 网络2::模型大小:7306KB,                       7.13MB,Size:64~256
normal 网络3::模型大小:29126KB,                     28.4MB,Size:128~512
normal 网络4::模型大小:120329KB,                   117MB,Size:128~1024
normal 网络5::模型大小:67176KB,                     66.1MB,Size:96~768
small 网络::模型大小:16400KB,                          16MB,   Size:96~384
small 网络1::模型大小:65474KB,                         63.9MB,Size:192~768

排序:
extra_large的Unet网络结构:模型大小:1944756KB,1.85GB,Size:16~4096
large Unet网络2::模型大小:486257KB,               474MB,Size:32~2048
large Unet网络::模型大小:121329KB,                 118MB,Size:64~1024
large Unet网络1:模型大小:121652KB,                  118MB,Size:16~1024
normal 网络4::模型大小:120329KB,                   117MB,Size:128~1024
normal 网络3::模型大小:29126KB,                     28.4MB,Size:128~512
normal网络::模型大小:30126KB,                        29.4MB,Size:64~512
normal 网络2::模型大小:7306KB,                       7.13MB,Size:64~256
small 网络1::模型大小:65474KB,                        63.9MB,Size:192~768
normal 网络5::模型大小:67176KB,                     66.1MB,Size:96~768
small 网络::模型大小:16400KB,                          16MB,   Size:96~384

按照大小排序:
extra_large的Unet网络结构:模型大小:1944756KB,1.85GB,Size:16~4096
large Unet网络2::模型大小:486257KB,               474MB,Size:32~2048
large Unet网络::模型大小:121329KB,                 118MB,Size:64~1024   ---- large net
large Unet网络1:模型大小:121652KB,                  118MB,Size:16~1024
normal 网络4::模型大小:120329KB,                   117MB,Size:128~1024
small 网络1::模型大小:65474KB,                        63.9MB,Size:192~768
normal 网络5::模型大小:67176KB,                     66.1MB,Size:96~768   ---- normal net
normal网络::模型大小:30126KB,                        29.4MB,Size:64~512   ---- small net
normal 网络3::模型大小:29126KB,                     28.4MB,Size:128~512
small 网络::模型大小:16400KB,                          16MB,   Size:96~384
normal 网络2::模型大小:7306KB,                       7.13MB,Size:64~256




回复 支持 反对

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则


Python|Opencv|MATLAB|Halcom.cn  

GMT+8, 2019-7-17 13:36 , Processed in 0.145635 second(s), 21 queries .

Powered by Discuz! X3.2

© 2001-2013 Comsenz Inc.

快速回复 返回顶部 返回列表