上传者: 38499732
|
上传时间: 2021-09-17 14:21:49
|
文件大小: 89KB
|
文件类型: PDF
Unet的网络结构:
根据该结构,用Pytorch实现Unet:
# -*- coding: utf-8 -*-
import torch
import torch.nn as nn
from torch.autograd import Variable
import torchvision.datasets as dset
import torchvision.transforms as transforms
import torch.nn.functional as F
import numpy as np
import torch.utils.data as Data
seed =