- 博客(2)
- 收藏
- 关注
RSS订阅原创 Unet网络搭建（Pytorch）
Unet是一个经典的语义分割网络，常常被用于医学影像的分割。在Unet的网络结构中，可以分为卷积模块，下采样模块以及上采样模块，详见下面的网络结构图： 在网络的搭建过程中，也是依照分为三大块这种思路进行搭建。话不多说，直接上代码：import torchimport torch.nn as nnimport torch.nn.functional as Ffrom torch.utils.tensorboard import SummaryWriterclass conv_block(nn.M
2022-02-27 10:51:23
3834
1
原创 DCGAN (Pytorch)
import torchimport torch.nn as nnimport torchvision.datasets as datasetsimport torchvision.transforms as transformsfrom torch.utils.data import DataLoaderfrom torch.utils.tensorboard import SummaryWriterfrom model import Discriminator,Generator,init.
2022-02-07 18:37:19
1192
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人