Skip to main content

Module assign

Module assign 

Source
Expand description

Simplified TaskAlignedAssigner (CPU-side).

For each GT box, alignment scores are: score = cls_score ^ alpha * iou ^ beta

The top-k anchors per GT are selected as positives. Conflicts (multiple GTs assigning the same anchor) are broken by highest alignment score.

Structsยง

AssignResult
Result of assigning GT targets to anchors for a single image.
TaskAlignedAssigner
TaskAlignedAssigner parameters.