原始的代码:
链接
我修改后的代码: 链接
1、env 环境的安装
windows10 64 bits、nvidia驱动、CUDA8.0、cudnn、anaconda
打开命令窗口, 分别输入以下指令:
conda create -n NestedUnetTorch python=3.6
conda activate NestedUnetTorch
pip install simpleitk
pip install opencv-python==3.4.2.16
pip install scipy
pip install scikit-learn==0.20
pip install scikit-image==0.14
conda install numpy mkl cffi
安装pytorch,选择与cuda版本对应的进行安装,参考链接如下:
链接
conda install torchvision -c pytorch
conda install Pillow=6.1
conda install tqdm
报错: ImportError: cannot import name '
_validate_lengths
', 由于numpy版本过高,升级下
pip install -U scikit-image
即可
conda install pandas
2、datasets 准备数据
脑肿瘤数据集
BraTs2018
链接:https://pan.baidu.com/s/1Ry41OVl9VLOM