Ei SSKrSSKJr SSKJrJr S S\R S\R S\S \S \R 4 S jjr SS\R S\R S \S \ \R \R 44S jjr g)N)_log_api_usage_once)_loss_inter_union_upcast_non_floatboxes1boxes2 reductionepsreturnc[RR5(d2[RR5(d[ [ 5 [ U5n[ U5n[XU5upEUS:XaU$US:Xa;UR5S:aUR5nU$SUR5-nU$US:XaUR5nU$[SUS35e)ac Gradient-friendly IoU loss with an additional penalty that is non-zero when the distance between boxes' centers isn't zero. Indeed, for two exactly overlapping boxes, the distance IoU is the same as the IoU loss. This loss is symmetric, so the boxes1 and boxes2 arguments are interchangeable. Both sets of boxes are expected to be in ``(x1, y1, x2, y2)`` format with ``0 <= x1 < x2`` and ``0 <= y1 < y2``, and The two boxes should have the same dimensions. Args: boxes1 (Tensor[N, 4]): first set of boxes boxes2 (Tensor[N, 4]): second set of boxes reduction (string, optional): Specifies the reduction to apply to the output: ``'none'`` | ``'mean'`` | ``'sum'``. ``'none'``: No reduction will be applied to the output. ``'mean'``: The output will be averaged. ``'sum'``: The output will be summed. Default: ``'none'`` eps (float, optional): small number to prevent division by zero. Default: 1e-7 Returns: Tensor: Loss tensor with the reduction option applied. Reference: Zhaohui Zheng et al.: Distance Intersection over Union Loss: https://arxiv.org/abs/1911.08287 nonemeanrgsumz$Invalid Value for arg 'reduction': 'z3 Supported reduction modes: 'none', 'mean', 'sum') torchjit is_scripting is_tracingrdistance_box_iou_lossr_diou_iou_lossnumelrr ValueError)rr r r loss_s V/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torchvision/ops/diou_loss.pyrrsF 99 ! ! # #EII,@,@,B,B12 v &F v &FVS1GDF  K f "jjlQ.tyy{ K58$((*4D K e xxz K29+=q r  c[X5up4X4U-- nURSS9upgpURSS9upp[R"Xj5n[R"X{5n[R"X5n[R"X5nUU- S-UU- S--U-nX-S- nX-S- nX-S- nX-S- nUU- S-UU- S--nSU- UU- -nUU4$)N)dimrr)runbindrminmax)rr r intsctunionioux1y1x2y2x1gy1gx2gy2gxc1yc1xc2yc2diagonal_distance_squaredx_py_px_gy_gcenters_distance_squaredrs rrr@s  &f5MF CK C]]r]*NBB2.Cc ))B C ))B C ))B C ))B C"%)!1sSyQ6F G# M 7a-C 7a-C 9/C 9/C!$sq 0cCiA5EF s7.1JJ KD 9r)rHz>)r8) rutilsr_utilsrrTensorstrfloatrtuplerrrr@s '8  6 LL6 LL66  6  \\ 6x LL LL  5<< %& r