search

ZhouL-lab / PLHN

15 stars 0 forks source link

代码运行 #1

Open RY-97 opened 2 weeks ago

RY-97 commented 2 weeks ago

感谢你的有意义的工作，但是你的代码运行过程，报错。我确保了我的类别数一致且正确 image

Zzzyyz1 commented 2 weeks ago

感谢您对我们工作的复现，我们检查了代码是可运行的，建议您可以打印您的输入来检查您的类别是否一致，也可能是monai库对于loss的计算有一些问题，同时可以检查一下代码运行环境。或者可以使用提供的公开数据集进行测试。

HQQ1224 commented 1 week ago

感谢你的有意义的工作，但是在我的运行main.py过程中出现了错误，在所提供的源码文件中找不到train5.txt文件。还有在Download the model weights 中没有找到latest_model.pth，导致运行test.py时报错。可以的话希望您能提供一些帮助。

6C032EEB24C724E2EEEAB2161B2A8F8C
Zzzyyz1 commented 1 week ago

感谢您对我们工作的复现。 1.如果您需要运行main.py来训练的话，train5.txt是训练集中数据的名称，源代码中的提供test3.txt中包含了YUNNAN公共数据集中的100个数据，您可以根据你的需要来划分trian和test，如果您要训练您自己的数据集的话可以按照格式自己生成txt文件。 2.关于test.py中的latest_model.pth，您可以从我们公开的权重中选择您需要测试的网络模型权重进行推理。