[Pytorch]는 UNET 의료 이미지 분할 시맨틱 UNET 코드는 (데이터의 복수의 세트를 포함)

A, UNET

UNET 의료 이미지 의미 분할의 초석이라고 할 수 있고, 종종 다른 모델 기준 비교. 일반 장기 분할은 종종만큼 더 분명 긍정적 정상적인 경계, 아무것도 특별한 상황으로, 일반적인 UNET은 잘 할 수있을 것입니다.

다음 UNET가 구성된다 :

UNET 특징 :

FCN 1.UNET 구조와 매우 유사하지만, 두 곳에 사이의 큰 차이는 점퍼 (연결 스킵) UNET 용으로 사용되는 접속 , 즉 서로 접합 채널 치수 특징 UNET을 사용 모드, 두껍게 형성 그것은 기능. 그리고 융합 FCN에 사용되는 해당 지점 추가 , 두꺼운 기능을 형성하지 않는다

2.UNET 네트워크 파라미터를 변경할 수있는 개념이다. 입력 화상은 일반적 5 UNET 다운 샘플링 후, 각각의 해상도는 일반적으로 감소 된 것으로,이 단계는 위상 인코더라고한다. 이어서도 5의 해상도가 두 배 것으로,이 단계는 디코더 단이라고있어서 시료 (또는 컨벌루션하여 보간) 내지 5 시간이있다. 중간 단계는 내가 제거하거나 단계 건너 뛰기 연결 층을 추가하려고, 다음 회선이 UNET 효과가 매우 가난하게, 연결을 생략하는 것이 중요하다.

 

두, UNET ++

알고 거의 명확 도입 된대로, 내가 ++ UNET 생각 ++ UNET 소개 : https://zhuanlan.zhihu.com/p/44958351

다음과 같이 UNET은 ++ 구성되어있다 :

 

세,주의-UNET

주의 - UNET 다음과 같은 구조 :

전송률 정보를 사용하여, 스티치 해당 기능의 해상도에 각각 인코더 및 디코더는, 인코더의 출력 특성을 재조정 상기 전. 도 1에 도시 된 바와 같이, 모듈은 위치 제어, 적색 원의 상이한 공간 특성의 중요성에 대한 게이트 신호를 생성한다.

주의 게이트 (절대 전송률는) 다음과 같은 구조를 가지고 :

AGs由有2个输入,如下图所示,分别是g和xl。g是较低分辨率的输入,如上图的绿色箭头。g个xl进入AGs前,g会先被上采样到跟xl分辨率相同。然后g和xl都会经过卷积核为1x1步长为1的卷积操作。1x1的卷积首先有降维的作用(降低通道维度),然后能让通道特征跨维度融合。经过激活函数ReLu增加非线性性,ReLu后的1x1卷积后主要作用是降维,把通道数降为1,然后激活函数sigmoid负责把特征图里的值映射到0~1之间,这时就得到注意力特征图了,最后一步就是把这个注意力特征与原来的xl相乘。得到最后的特征图。综合而言就是使用拥有较高语义特征的特征图来指导当前特征图进行注意力的选择

四,R2U-NET

在R2U-NET论文中,此网络被用于眼球血管分割和皮肤癌分割,肺部分割任务中。R2U-NET全称为基于UNET的循环残差神经网络。

R2U-NET的结构如下:

上图中,有圈的地方就是R2U-NET中的循环结构,R2U-NET就是在UNET中嵌入了这种循环结构的网络,该循环结构详细说明如下图:

(a)为普通的两个conv模块,(b)为使用了循环卷积的模块,(c)为使用了残差卷积的模块,(d)是同时使用了残差卷积和循环卷积的模块

一个循环卷积模块其实就是t(t是可以自己设置的超参数)个普通的conv(conv+BN+ReLU)的堆叠,只不过,除了第一个conv的输入是x(输入图片)之外,后边的conv的输入是前一个conv的输出加上x作为输入。

 

五,CENET

CENET的结构如下:

CENET的编码器用的是resnet34。中间部分则由作者提供的DAC和RMP模块组成,DAC和RMP模块等下会有介绍。

下图中,左边的是DAC模块。可以看出作者的DAC模块很大程度地参考了inception模块,通过提供不同尺度的卷积核来获得多尺度的空间信息。感受野大的卷积操作可以为大目标提取和生成更抽象的特征,而小感受野的卷积对于小目标则更好。而右边的则是RMP模块,RMP模块由于是通过连接不同池化操作生产的特征图,由此RMP并无额外增加参数,在作者的文章中说,RMP主要用于获取上下文信息,但是我觉得这作用不打,因为医学图像的上下文信息个人认为不是十分的重要,而我在DRIVE数据集中,把RMP去掉确实影响不是十分大。

 

代码:

本人通过整合以上5个模型代码+FCN和SegNet,而且有7个数据集的链接,整合了一套用于基于UNET及其变体的医学图像代码。其中包括了指标可视化功能和预测结构可视化功能:

代码链接:https://github.com/Andy-zhujunwen/UNET-ZOO

发布了265 篇原创文章 · 获赞 392 · 访问量 74万+

추천

출처blog.csdn.net/u014453898/article/details/104271746