模型问题

[niugie]

尊敬的大佬您好，我想直接把您的模型下载下来然后直接预测数据集中的数据得出去云结果。其中cuhk1和cuhk2成功，但是另外两个数据集一直失败，我想问一下如何调整代码，只需要修改配置文件中的模型路径和数据集路径吗？是否需要修改其他代码的内容。恳求解答，非常感激。

[Ly403]

您好，十分感谢您对我们的论文感兴趣。

关于您的问题，我想只需要修改一下模型路径和数据集路径应该是可以跑通的，当然前提是您按照readme文件配置好的环境。如果还是遇到问题，您可以将您遇到问题的截图贴出来，我会努力尝试解决您的问题。

[niugie]

大佬您看，我是设置好对应的虚拟环境和包，运行命令python main.py --base configs/example_training/sen2_mtc_new.yaml --enable_tf32 -t false报的错误。

运行信息.txt

[Ly403]

您需要修改yaml文件里面model.params.input_key这个键对应的值，我根据您提供的运行信息看到现在这个键的值被设置为label。然而代码并没有从batch字典里面读取到label键，这个batch字典是由data module类返回的数据，可以参考这一行。您需要将input_key设置为target图片对应的键，如果您没有修改sen2_mtc_new数据类的代码，那就应该是gt_image。如果您修改了代码，那应该设定为您返回的gt图像对应的键的名称。