FPh jddlmZddlZddlmZmZddlmZddlm Z ddl m Z e e dZ d d Z d Zd Zy) )partialN)IterableSimpleNamespace yaml_load) check_yaml)BOTSORT) BYTETracker bytetrackbotsortc~t|dr|ryt|jj}t dit |}|j dvsJd|j dg}t|jjD]-}t|j |d}|j|/||_ y) ab Initialize trackers for object tracking during prediction. Args: predictor (object): The predictor object to initialize trackers for. persist (bool, optional): Whether to persist the trackers if they already exist. Defaults to False. Raises: AssertionError: If the tracker_type is not 'bytetrack' or 'botsort'. trackersNr z9Only support 'bytetrack' and 'botsort' for now, but got '')args frame_rate) hasattrrrtrackerrr tracker_typerangedatasetbs TRACKER_MAPappendr) predictorpersistrcfgr_s eC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\ultralytics/trackers/track.pyon_predict_startr!sy*%'//0G ! 7Ig$6 7C   7 7X CCDTDTCUUVWX 7H 9$$'' (c../SRH )"Ic D|jj}|jd}t|D]}|j|j j j}t|dk(rG|j|j|||}t|dk(rx|dddfjt}|j|||j|<|j|jtj|ddddfy)z;Postprocess detected boxes and update with object tracking.rrN)boxes)rrbatchrresultsr%cpunumpylenrupdateastypeinttorch as_tensor)rrim0sidettracksidxs r on_predict_postprocess_endr5(s     B ??1 D 2Y"((,,.446 s8q= ##A&--c47; v;!  QUm""3'(003C8 !!##%//&CRC.*I#Jr"cr|jdtt||jdty)a Register 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. r!)rr5N) add_callbackrr!r5)modelrs r register_trackerr98s/ )73CW+UV 35OPr")F) functoolsrr.ultralytics.utilsrrultralytics.utils.checksrbot_sortr byte_trackerr rr!r5r9rr"r r?s4 @/%'G< "0 K Qr"