SSD 多尺度连接问题
详细解释 就是将一个通道的拉起来排列开来
def flatten_pred(pred):
return torch.flatten(pred.permute(0, 2, 3, 1), start_dim=1)
def concat_preds(preds 0
点赞
收藏
分享 
微信扫一扫
Linux网络相关内容与端口
相关推荐
0 条评论
微信扫一扫
详细解释 就是将一个通道的拉起来排列开来
def flatten_pred(pred):
return torch.flatten(pred.permute(0, 2, 3, 1), start_dim=1)
def concat_preds(preds相关推荐