search

Haochen-Wang409 / U2PL

[CVPR'22] Semi-Supervised Semantic Segmentation Using Unreliable Pseudo-Labels
Apache License 2.0
426 stars 59 forks source link

About checkout model of cityscapes #97

Closed MRtianyanxiaobai closed 1 year ago

MRtianyanxiaobai commented 1 year ago

作者您好，我在使用您所给的checkout时遇见了问题。我将您给的预训练模型 city_744_AEL_U2PL_78.51.pth 放在了 U2PL/experiments/cityscapes/744/ours/checkpoints 下，并且修改名字为 ckpt.pth 然后运行出现错误如下，应该是预训练模型中没有这两个字段： image

Haochen-Wang409 commented 1 year ago

您可以在 这一行 之前加一句： checkpoint = {'model_state': checkpoint}

MRtianyanxiaobai commented 1 year ago

ok，问题已解决，谢谢作者大大