![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch+Unet
文章平均质量分 64
lryjy
这个作者很懒,什么都没留下…
展开
-
nnUnet(代码)-训练部分
学习目标:逐步分析nnunet训练部分学习内容:training部分· 拿到训练plans(计划)· 初始化数据增强参数· 采用五折交叉验证· dataset与dataloader/数据加载过程· 初始化网络· 初始化优化器与学习率函数1.nnUNetTrainer(版本一的训练方法)··· 损失函数:self.loss = DC_and_CE_loss({'batch_dice': self.batch_dice, 'smooth': 1e-5, 'do_bg': False}, {}原创 2021-04-30 16:06:45 · 3243 阅读 · 0 评论 -
入门Unet:自己总结方便后面回看
入门Unet网络学习第一部分:网络的结构Unet_part.py:目的:实现Unet网络中需要用到的几个基础类:大部分的解释在代码中:import torchimport torch.nn.functional as Fclass DoubleConv(torch.nn.Module): def __init__(self, in_ch, out_ch): super(DoubleConv, self).__init__() self.double原创 2020-10-16 00:00:43 · 2275 阅读 · 5 评论