cornernet------loss

首先我们使用网络训练出左上角的heatmap_tl以及右下角的hetmap_br,这个heatmap维度为(batch,128,128,80)其中80是类别数。在制作数据集的时候,将数据也同样制作成了(batch,128,128,80)。以左上角为例,我们每一类都有一个(128,128)的矩阵与之对应。输入网络一张图片,我们会得到左上角以及右下角的heatmap,这个heatmap里面又包括了每一
本站公众号
   欢迎关注本站公众号,获取更多信息