implus / GFocalV2

Generalized Focal Loss V2: Learning Reliable Localization Quality Estimation for Dense Object Detection, CVPR2021

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

regression_target分布统计问题?

johnson-magic opened this issue · comments

@implus
如题,按照我的理解,我将训练集1个epoch的regression_target保存了下来(shuffle设定为False),并可视化结果如下图所示。
image
感觉和论文中翔哥你展示的不一致。
image

我的做法是在gfl_head.py找那个将target_corners这个变量保存下来,然后画直方图,请问有什么问题吗?

with open('./target_corners_results/{}.npy'.format(time.time()), 'wb') as f:
        np.save(f, target_corners.cpu().numpy())