DeepLung训练LUNA16数据报错

Question

Ubuntu22.04使用终端跑https://github.com/Minerva-J/DeepLung（预测肺结节），在完成LUNA16胸腔CT图片预处理后进行训练时报错

错误提示为：

cannot reshape array of size 12038144 into shape (1,241,215,279)
[Errno 104] Connectin reset by peer

训练代码如下：

python main.py --model dpn3d26 -b 8 --save-dir dpn3d26/retrft960/ --epochs 100 --config config_training0

运行结果如下图：

不确定是因为CT图片预处理过程出了问题，还是因为deeplung虚拟环境设置有误，或者是代码问题

请各位帮忙解决一下这个问题

Answer 1

这个问题可能是由于训练数据的大小不匹配造成的。检查下训练数据是否与模型的输入大小匹配，并确保代码正确地处理了训练数据。