Expand description
YOLO26 training loss: CIoU box regression + BCE classification.
§Pipeline (per training step)
- Decode raw LTRB predictions to XYWH using the anchor grid.
- Assign GT boxes to anchors (CPU-side TaskAlignedAssigner).
- Box loss: CIoU forward (saved activations) then backward.
- Cls loss: BCE cls backward on all anchors.
- Decode backward: d_xywh → d_ltrb through the linear decode.
- Return
d_boxes[B, 4*A]andd_scores[B, nc*A](host f32).
Re-exports§
pub use super::anchor::AnchorGrid;pub use super::assign::TaskAlignedAssigner;
Structs§
- Yolo26
Loss - CUDA training loss for YOLO26: computes d_boxes and d_scores.
Functions§
- ciou_
forward_ cpu - CPU reference CIoU forward — returns saved activations (iou, v, alpha).