github上把项目下载下来并运行
GitHub上把项目下载下来
本例是以UNET-ZOO项目为例,
- 查看项目中的README文件,win10、pycharm、python3.6、pytorch1.3.1
- 把README中提到的数据集下载下来。
从GitHub上下载项目到本地的工作位置,
本机是D:\Anaconda3\envs\UNET-ZOO\UNET-ZOO
Pytorch环境的搭建
1、创建虚拟环境python 3.6,版本高点也是可以,的虚拟环境
conda create -n UNET-ZOO python=3.6
2、激活UNET-ZOO环境并查看python的版本是否为3.6
conda activate UNET-ZOO
python -V
安装Pytorch,从官网下载pytorch 1.3.1
选择pytorch的版本,本机测试的是pytorch 1.9.1版本,具说pytorch 2.0也可以。
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=10.2 -c pytorch
在PyCharm编译环境中关联UNET-ZOO虚拟环境
用PyCharm打开下载下来的UNET-ZOO项目,然后关联虚拟环境;
下载数据集
按照README上的提示下载相应的数据集,放到项目内。
将下载live数据集放到项目目录下。
根据README的操作方法修改datasets.py
右击liver文件夹下的train,然后选择复制路径,将dataset.py中的代码路径改掉,其中val文件夹也是同样的操作。
开始训练
复帛README中的运行指令python main.py --action train&test --arch UNet --epoch 21 --batch_size 21
由于是训练和测试一块的,所以要删除掉&test
项,注:在命令行中执行时要以自己创建的UNET-ZOO身份才是自己的虚拟环境。
缺失的库需要自行安装
如:sklearn库
pip install scikit-learn
pip install scikit-image
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python