如何在此输入和目标上使用置换？

Posted 2023-03-27

【中文标题】如何在此输入和目标上使用置换？

【英文标题】：How to use permute on this Input and Target?

【发布时间】：2020-10-02 12:39:50

【问题描述】：

我的语义分割掩码有 5 个类 + 1 个（背景）错误。

如何使用 permute 来避免这种情况？

Target size (torch.Size([4, 1, 320, 480, 6])) must be the same as input size (torch.Size([4, 6, 320, 480]))

【参考方案1】：

你可以结合permute和unsqueeze：

import torch

x = torch.rand((4, 6, 320, 480))
new_x = x.permute((0,2,3,1)).unsqueeze(1)
# new_x.shape = torch.Size([4, 1, 320, 480, 6])