ultralytics-TaskAlignedAssigner
·
TaskAlignedAssigner
结合PD与GT动态生成target信息。
select_candidates_in_gts
如果anchor的中心坐标与gt的左上角或右下角的差值小于阈值则设为positive anchor,得到表达positive anchor的mask_in_gts。
get_box_metrics
通过GT与PD计算iou与分类score的联合metrics。
select_topk_candidates
基于iou与分类score的联合metrics选择top-k候选框得到mask_topk。
选择的过程中遇到某个anchor成为多个GT的候选框的时候, 过滤该anchor。
get_pos_mask
mask_pos = mask_topk * mask_in_gts * mask_gt
mask_gt是初始标注信息[bs,max_nums,1],根据box数值是否存在判定有效目标位置。
select_highest_overlaps
如果某个anchor成为多个GT的候选框的时候,选择iou最大的。更新mask_pos.
TaskAlignedAssigner.forward
通过更新后的mask_pos更新metrics,再对metrics做最大值归一化,归一化后的数值作为iou的权重相乘得到新的metrics。
更多推荐
已为社区贡献1条内容
所有评论(0)