运行TransUnet踩坑记录
运行TransUnet踩坑记录
-
- 数据集制作问题
- 在运行train.py过程中遇到的问题
数据集制作问题
数据集制作要用到一些*.py文件，基本参照这篇博主的记录，就能制作自己的https://blog.csdn.net/qq_37652891/article/details/123465472?spm=1001.2014.3001.5501
在运行train.py过程中遇到的问题
1.首先要装几个模块:ml-collections，这个都简单。第二个模块叫做medpy，在安装他的过程中要一并安装SimpleITK包，用pip安装，一碰到这个包，就弹出了关于编译器cl的错误，没办法安装成，起初以为是环境问题，还重新创建了新的环境，但是没有什么作用。后来通过 conda install -c simpleitk simpleitk 没有经过编译就成功了。至于第一次用pip安装出现的编译错误，始终不知道原因。
2.第二个问题是trainer.py报错：ModuleNotFoundError: No module named ‘datasets.dataset_synapse’，这个问题在网上看到解决方法，具体原因不明，好像是什么python根目录和项目目录问题。
解决方法：在trainer.py import 行代码添加
current_dir = os.path.abspath(os.path.dirname(r'G:\codelearn\papercode329\2TransUNet-main)'))
sys.path.append(current_dir)
import dataset_synapse
绿色的是自己的项目路径，再注释掉下面报错的行
# from datasets.dataset_synapse import Synapse_dataset, RandomGenerator
3.接下来再运行看到了进度条，又报错：
RuntimeError: CUDA error: device-side assert triggered
查找了一番，后来得到了灵感，原因是因为我要分割的总共4类，但是代码中填写的类别数还要加上背景，于是填写num_class=5。