pifpaf的loss生成

class CompositeLossByComponent(torch.nn.Module):
"""Default loss until v0.12"""这是默认的损失函数,之后作者进行版本更新对其进行了舍弃。

在forward中把输入分为confiden,reg,scale这三个特征图然后放入_confidence_loss和_location_loss和_scale_loss。最后三个损失做总和。

 

posted @ 2023-03-07 16:15  祥瑞哈哈哈  阅读(20)  评论(0)    收藏  举报