jhb86253817 / PromptMRG

MIT License
50 stars 6 forks source link

权重文件是哪个数据集训练出来的 #1

Closed deng1604 closed 6 months ago

deng1604 commented 8 months ago

作者你在介绍时说: 或者,您可以从此处下载经过训练的模型权重。请注意,此模型权重是使用 R2Gen 的图像训练的。如果您使用自己处理的图像,则此权重可能会降低性能。在这种情况下,您需要自己训练模型。

请问,您给出的模型的权重文件,是MIMIC数据集的还是IU Xray数据集对应的权重文件呢

deng1604 commented 8 months ago

另外,缺少训练train_iu_xray.sh文件。测试时,--load_pretrained results/promptmrg/model_best.pth 这个文件也缺少,作者可以提供下吗,不胜感激!

jhb86253817 commented 8 months ago

你好,我给出的权重是在MIMIC上训练的。我们没有在IU上训练,而是直接用MIMIC训的模型在IU上测。更多细节可以从论文里找到,谢谢~

deng1604 commented 8 months ago

你好,我给出的权重是在MIMIC上训练的。我们没有在IU上训练,而是直接用MIMIC训的模型在IU上测。更多细节可以从论文里找到,谢谢~

好的,谢谢你❤

bmyoon1234 commented 7 months ago

Is it possible that the MIMIC data downloaded from R2gen contains images processed differently from the MiMIC images provided on the physionet site? Or is it just the MIMIC annotation files?

jhb86253817 commented 7 months ago

Is it possible that the MIMIC data downloaded from R2gen contains images processed differently from the MiMIC images provided on the physionet site? Or is it just the MIMIC annotation files?

Yes, I think the images from R2Gen has been processed, which is slightly different from the images of physionet. The trained model weight I provided is trained on R2Gen images.