Centernet的HearMap绘制及改进
作者：互联网
Centernet的heatmap绘制及改进
Centernet
Centernet是经典的Anchor-free目标检测算法：
1.一种是基于三元组（左上角点，中心点，右下角点）的Centernet；
2.另外一种是只基于中心点，利用heatmap(热图)进行训练
Heatmap
Heatmap的绘制分一下几点：
-
创建0矩阵，num_classes为数据集中物体的类别数，如COCO是80类，DOTA-1.0是15类；
-
计算中心点 ，GroundTruth计算中心点位置（Centernet算法先仿射变换成原图像的1/4，所以GroundTruth也需要进行相应的转换）；
-
Heatmap的绘制，Centernet的热图是绘制的圆的高斯分布，但我们若想绘制一个旋转平移椭圆的高斯分布，可以采用先对椭圆先进行平移变换，再基于cv.getRotationMatrix2D（）与cv.warpAffine（）完成椭圆旋转。
标签：椭圆,Centernet,Heatmap,HearMap,GroundTruth,绘制,高斯分布 来源： https://blog.csdn.net/qq_43807997/article/details/113408866