感谢【呼呼】同学分享的“BraTS2017数据集的迁移学习+深度学习提取特征+机器学习”
迁移学习
一、迁移学习(把之前对Medicalnet的迁移学习内容跑了一遍,用的BraTS2017的数据集) 1、下载Med3D源码,完成基础运行环境配置 (1)https://github.com/Tencent/MedicalNet,把pretrain的部分(包含data,pretrian、trails三个文件)一起下载下来,都放在MedicalNet_master的文件夹里; (2)从pycharm中打开MedicalNet_master,在settings的解释器部分选择python;
2、跑通Demo和训练模型 (1)在Ternimal中激活onekey:conda activate onekey; (2)运行模型:python train.py --gpu_id 0(#GPU=1) (3)因为onekey和MedicalNet_master中都有dataset和utils,所以要纠正2个moduleNotFoundError,将MedicalNet_master中的重命名为dataset_med和utils_med,同时在train.py中进行相应修改;</