SSD 多尺度连接问题

SSD 多尺度连接问题

详细解释 就是将一个通道的拉起来排列开来

def flatten_pred(pred):
    return torch.flatten(pred.permute(0, 2, 3, 1), start_dim=1)

def concat_preds(preds

你可能感兴趣的:(AI论文写作,python,pytorch,深度学习)