FiddlmZddlmZddlZddlmZmZddlm Z ddl m Z ddl m Z e e d Zdd ed ed dfd Zdd ed ed dfdZded ed dfdZy))partial)PathN)YAMLIterableSimpleNamespace) check_yaml)BOTSORT) BYTETracker) bytetrackbotsort predictorpersistreturncjjdk(r tdtdr|ryt jj }t ditj|}|jdvrtd|jdd_ tdrjj|jd k(r|jr|jd k(rd d lm}t%jjt&j(j*rZt%jjjd |r-jjjd j,rd|_n<fd}jjjd j/|_ g}t1j2j4D]H}t7|j|d}|j9|j2j:dk7sHn|_dgj2j4z_y)aInitialize trackers for object tracking during prediction. Args: predictor (ultralytics.engine.predictor.BasePredictor): The predictor object to initialize trackers for. persist (bool, optional): Whether to persist the trackers if they already exist. Examples: Initialize trackers for a predictor object >>> predictor = SomePredictorClass() >>> on_predict_start(predictor, persist=True) classifyu/❌ Classification doesn't support 'mode=track'trackersN>r r z?Only 'bytetrack' and 'botsort' are supported for now, but got ''_hookr autor)Detectzyolo26n-cls.ptc,t|d_y)Nr)list_feats)moduleinputr s \/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/trackers/track.pypre_hookz"on_predict_start..pre_hook8s#'a>  )args frame_ratestream) r!task ValueErrorhasattrrtrackerrrload tracker_typeAssertionErrorrrremove with_reidmodelultralytics.nn.modules.headr isinstancetorchnnModuleend2endregister_forward_pre_hookrangedatasetbs TRACKER_MAPappendmodervid_path)r rr(cfgrrr_s` ron_predict_startr?s~~j(JKKy*%'//0G ! 7DIIg$6 7C 77^_b_o_o^ppqrssIy'"  9$399;N6 y,,ehhoo >9??0066r:FCOO))//3;;(CI 2(oo3399"=WWX`aIOH 9$$'' (c../SRH    ! !X -   "I)"3"3"6"66Irc V|jjdk(}|jjdk(}t |j D]^\}}|j |r|nd}|jt|jjz }|s9|j|r|nd|k7r#|j||j|r|nd<|r |jn |jjj!}|j#||j$t'|dd} t)| dk(r| dddfj+t,} || |j |<|rdndt/j0| ddddfi} |j |j"di| ay) aPostprocess detected boxes and update with object tracking. Args: predictor (object): The predictor object containing the predictions. persist (bool, optional): Whether to persist the trackers if they already exist. Examples: Postprocess predictions and update with tracking >>> predictor = YourPredictorClass() >>> on_predict_postprocess_end(predictor, persist=True) obbr#rfeatsNrboxesr$)r!r%r7r; enumerateresultsrsave_dirrpathnamer<resetrArCcpunumpyupdateorig_imggetattrlenastypeintr1 as_tensor) r ris_obb is_streamiresultr(r<dettracksidx update_argss ron_predict_postprocess_endr[Gst^^ E )F!!&&(2Iy0013 6$$)Q;%%V[[(9(>(>>9--9a!DP MMO8@I  Iq1 5#vzz::<BBDV__gfgt6TU v;!  QUm""3'%c{ ! &uGU__VAsPRsF^5TU # !##2k23rr.c|jdtt||jdtt|y)aRegister tracking callbacks to the model for object tracking during prediction. Args: model (object): The model object to register tracking callbacks for. persist (bool): Whether to persist the trackers if they already exist. Examples: Register tracking callbacks to a YOLO model >>> model = YOLOModel() >>> register_tracker(model, persist=True) r?)rr[N) add_callbackrr?r[)r.rs rregister_trackerr^gs7 )73CW+UV 3W=Wah5ijr)F) functoolsrpathlibrr1ultralytics.utilsrrultralytics.utils.checksrbot_sortr byte_trackerr r9objectboolr?r[r^r$rrrgs| ;/%(G< 272727$27j3&343D3@ kF kT kd kr