Open liushh39 opened 1 year ago
十分感谢
你好,但是运行evaluate.py出现一下错误 RuntimeError: Error(s) in loading state_dict for BiSeNet: size mismatch for conv_out.conv_out.weight: copying a param with shape torch.Size([9, 128, 1, 1]) from checkpoint, the shape in current model is torch.Size([19, 128, 1, 1]). size mismatch for conv_out16.conv_out.weight: copying a param with shape torch.Size([9, 64, 1, 1]) from checkpoint, the shape in current model is torch.Size([19, 64, 1, 1]).
evaluate.py和相关数据集已经更新 你可以重新下载运行。
语义分割的模型已经上传至model_final.pth