![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
unet
guyuezunting
这个作者很懒,什么都没留下…
展开
-
window pytorch unet代码学习之random_split
先看看调用train, val = random_split(dataset, [n_train, n_val])def random_split(dataset, lengths): r""" Randomly split a dataset into non-overlapping new datasets of given lengths. Arguments: dataset (Dataset): Dataset to be split .原创 2020-07-14 17:47:00 · 2420 阅读 · 0 评论 -
windows pytorch unet网络学习之BasicDataset
dataset = BasicDataset(dir_img, dir_mask, img_scale)上面是调用语句class BasicDataset(Dataset): def __init__(self, imgs_dir, masks_dir, scale=1): self.imgs_dir = imgs_dir self.masks_dir = masks_dir self.scale = scale asser.原创 2020-07-13 22:39:01 · 1869 阅读 · 0 评论 -
windows pytoch 学习Unet代码解读之UNET类文件解读
首先看外部初始化调用代码# Change here to adapt to your data # n_channels=3 for RGB images # n_classes is the number of probabilities you want to get per pixel # - For 1 class and background, use n_classes=1 # - For 2 classes, use n_classes=1原创 2020-07-12 22:44:48 · 2223 阅读 · 1 评论 -
window 学习pytorch unet代码之self.inc = DoubleConv(n_channels, 64)
self.inc = DoubleConv(n_channels, 64)可以猜测,channels是输入channel,64是要输出的channel看看DoubleConv函数的具体实现class DoubleConv(nn.Module): """(convolution => [BN] => ReLU) * 2""" def __init__(self, in_channels, out_channels, mid_channels=None): .原创 2020-07-12 20:25:37 · 4598 阅读 · 0 评论 -
windows pytorch-unet网络学习一
为了用pytorch实现pointnet,需要先学习pytorch的一个网络,然后再进行实现关于pytorch的安装,个人建议初学者安装cpu的最好,因为gpu的pytorch在windows下特别难以安装成功因为gpu目前实在版本混乱,而且跟tensorflow又各有区别,而且pytorch支持的cuda和cudnn的版本有时候英伟达都下载不到了所以,强烈建议安装cpu的,速度慢点没关系,我们先入门了再说,要不然卡在环境搭建那里,真的非常浪费时间所以强烈建议去pytorch官网找最新版本c原创 2020-07-01 10:08:28 · 409 阅读 · 0 评论 -
windows pytorch Unet网络实现二,网络学习
学习代码之前先查看网络,有很多已有的图,都是标准的首先看输入是572x572的图片,所以,处理ct的时候需要先读取image这样的信息第一次卷积,输出570*570并且输出是64channel的数据。这里用的padding是valid,所以在3*3的核,stride1*1的时候,都需要valid的数据,因此,3*3只能移动到最后一列有效数据停下,刚刚好少2列2行,因此是570*570的。接着进行第二次卷积,将特征进一步提高。也是输出64channel数据,大小是568*568接着这是第一次.原创 2020-07-09 20:26:46 · 1013 阅读 · 0 评论