mmsegmentation中运行unet框架,encoder_decoder.py中出现下面问题 每4000次迭代之后,评估时出现的

img

img


第二张图就是要修改的部分,希望有兄弟能帮忙解决一下,困扰我一整天了都

h_img, h_crop等都是数据的size,应该都是二维三维的,不能直接跟后面的1做加减, 你可以尝试一下将 1 扩展成等维度的数据

报错意思就是不支持torch.size和int类型进行运算,你把1改成torch.ones((宽度,高度))试