Struct TaskAlignedAssigner 

pub struct TaskAlignedAssigner {
    pub top_k: usize,
    pub alpha: f32,
    pub beta: f32,
}

TaskAlignedAssigner parameters.

Fields

top_k: usize

Top-k anchors selected per GT.

alpha: f32

Exponent on the predicted class confidence.

beta: f32

Exponent on the predicted IoU.

Implementations

impl TaskAlignedAssigner

pub fn assign( &self, pred_xywh: &[f32], pred_scores: &[f32], anchor_cx: &[f32], anchor_cy: &[f32], gt_boxes: &[[f32; 4]], gt_cls: &[usize], ) -> AssignResult

Assign GT boxes to anchors for a single image.

Arguments

• pred_xywh – decoded anchor predictions [4, A] channels-first
• pred_scores – per-anchor class logits [nc, A] channels-first (sigmoid applied internally)
• anchor_cx – anchor centre x [A]
• anchor_cy – anchor centre y [A]
• gt_boxes – GT boxes [[cx, cy, w, h]; M]
• gt_cls – GT class indices [M]

Trait Implementations

impl Clone for TaskAlignedAssigner

fn clone(&self) -> TaskAlignedAssigner

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Default for TaskAlignedAssigner

fn default() -> Self

Returns the “default value” for a type.