Author: Zongwei Zhou | 周纵苇
Weibo: @MrGiovanni
Email: zongweiz@asu.edu
Please cite this paper if you found it useful. Thanks!
Wang H, Zhou Z, Li Y, et al. Comparison of machine learning methods for classifying mediastinal lymph node metastasis of non-small cell lung cancer from 18F-FDG PET/CT images[J]. 2017, 7.
参考文献.
- 周纵苇.基于深度学习的医学影像分类识别.大连理工大学本科毕业论文.2016.6
1. 数据扩充(MATLAB)
详细解说:
MATLAB处理大量三维图像,Data Augmentation
-
CrossValidation.m
描述:主函数
实现数据扩充,交叉验证,通过旋转,平移变换,得到不同的2.5D数据。
-
RotParGio.m
描述:子函数
输入:3d图像,沿x,y,z轴旋转的角度,patch的尺寸。
输出:2.5D的三个图[slide_x, slide_y, slide_z]。
-
iSaveX.m
描述:子函数
CrossValidation.m程序中用了MATLAB并行运算,而parfor中不能直接用save()函数,因此这儿定义一个函数作为过渡。
-
Zscore.m
描述:主函数
数据Z-Score处理,可做可不做,每个通道的数据减去平均值,再除以标准差。经验显示Z-Score后在CNN中训练的效果好。
Z-Score:x=(x-mean(x))/std(x).
2. 卷积神经网络(Python)
详细解说:
简易的深度学习框架Keras代码解析与应用
深度学习框架Theano利用GPU加速
-
cnn.py
描述:主函数
包含CNN网络的训练,测试的全过程。
-
data.py
描述:子函数
实现数据读入,输出的是正负样本等分后的训练集和测试集。
-
CNNsoftmax.py
描述:主函数
用SVM和RF代替softmax。
-
ROC_Matlab.py
描述:主函数
计算出置信度,存到文件中,再用MATLAB函数画ROC曲线。