github上把项目下载下来并运行

如何把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


转载请注明出处:  https://www.cntworld.cn
智能工控 » github上把项目下载下来并运行

发表回复

提供最优质的资源集合

立即查看 了解详情