简介
CSP: Center and Scale Prediction
方法
实验
代码
准备ground truth
1 | def calc_gt_center(C, img_data,r=2, down=4,scale='h',offset=True): |
seman_map有三个位面,第一个是高斯值mask,第二个是学习权重,第三个是目标中心点的位置。
网络结构
1 | def nn_p3p4p5(img_input=None, offset=True, num_scale=1, trainable=False): |
Loss
1 | def cls_center(y_true, y_pred): |