请问一下,如何成功的在自己的电脑上成功运行这个医疗影像分割模型啊,求十分具体的步骤,谢谢
https://github.com/HiLab-git/LCOVNet-and-KD

这是最后一次调试出现的问题,求指教
请问一下,如何成功的在自己的电脑上成功运行这个医疗影像分割模型啊,求十分具体的步骤,谢谢
https://github.com/HiLab-git/LCOVNet-and-KD

关注github仓库已经有对应步骤,
但是比较简单好多细节没有提到。
.
这边下载到本地试了下,补充了下具体需要操作的部分,
下面是需要的具体操作 以及最后 github上的操作步骤:
git clone https://github.com/HiLab-git/LCOVNet-and-KD.git
https://pan.baidu.com/share/init?surl=HwD1iqHorgXfYXnrChdzIg
9jlj
下载后 放到
teacher.model
teacher.model.pkl
拷贝到
LCOVNet-and-KD/for_tmi/KD/ 文件夹下
安装第三方库
对于CUDA 11
python3 -m pip install PYMIC torch torchvision torchaudio ml_collections einops timm monai -i https://pypi.tuna.tsinghua.edu.cn/simple
对于CUDA 10.1
python3 -m pip install torch==1.9.0+cu101 torchvision==0.10.0+cu101 torchaudio==0.9.0 ml_collections einops timm monai -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
/home/data/zhaoqianfei/new_pymic
全局替换为->
你的本地路径/LCOVNet-and-KD
cd LCOVNet-and-KD
python3 pymic/net_run/train.py for_tmi/KD/config/kd.cfg
补充:
1、安装PyMIC:
https://github.com/HiLab-git/PyMIC
python3 -m pip install PYMIC
并将文件添加到 Pymic。
2、 从百度网盘:
https://pan.baidu.com/share/init?surl=HwD1iqHorgXfYXnrChdzIg
下载预训练模型和示例CT图像(提取码9jlj)。
3、运行./KD/run.sh。结果将保存在./KD/model/kd.
如有问题及时沟通