AlibabaResearch / AlibabaResearch/AdvancedLiterateMachinery

LORE 这部分代码问题

未关闭
#186 0 条评论 0 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
C++
星标
1.8k
派生
195
PR 合并指标
30 天内没有已合并 PR

描述

wh labels计算的是:

wh[k] = ct[0] - 1. * CorNer[0], ct[1] - 1. * CorNer[1], \
ct[0] - 1. * CorNer[2], ct[1] - 1. * CorNer[3], \
ct[0] - 1. * CorNer[4], ct[1] - 1. * CorNer[5], \
ct[0] - 1. * CorNer[6], ct[1] - 1. * CorNer[7]
中心点减去四个角点的值,这个是宽高wh?

后面又通过 ind 求出中心点对应在特征图上的坐标xs,ys

再减去wh 得到是什么值?

def _get_wh_feat(ind, output, ttype):

width = output['hm'].shape[2]
xs = (ind % width).unsqueeze(2).int().float()
ys = (ind // width).unsqueeze(2).int().float()
if ttype == 'gt':
wh = output['wh']
elif ttype == 'pred':
wh = _tranpose_and_gather_feat(output['wh'], ind)
ct = torch.cat([xs, ys, xs, ys, xs, ys, xs, ys], dim=2)
bbx = ct - wh

return bbx

贡献指南

这个仓库没有索引到贡献指南

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。