a Dfe @sdZddlZGdddZdS)z Callback utils Nc@s:eZdZdZddZdddZddd Zd d d d ZdS) Callbacksz<" Handles all registered callbacks for YOLOv5 Hooks cCs8gggggggggggggggggggd|_d|_dS)N)on_pretrain_routine_starton_pretrain_routine_endon_train_starton_train_epoch_starton_train_batch_startZoptimizer_stepZon_before_zero_gradon_train_batch_endon_train_epoch_end on_val_starton_val_batch_starton_val_image_endon_val_batch_end on_val_endon_fit_epoch_end on_model_save on_train_endon_params_updateZteardownF) _callbacks stop_training)selfrE/data/NAS7/SABIOD/METHODE/ermites/yolo-dyni/yolov5/utils/callbacks.py__init__s*zCallbacks.__init__NcCsR||jvs Jd|d|jt|s8Jd|d|j|||ddS)z Register a new action to a callback hook Args: hook: The callback hook name to register the action to name: The name of the action for later reference callback: The callback to fire hook '' not found in callbacks z callback 'z' is not callable)namecallbackN)rcallableappend)rhookrrrrrregister_action&s zCallbacks.register_actioncCs|r|j|S|jS)z" Returns all the registered actions by callback hook Args: hook: The name of the hook to check, defaults to all )r)rr rrrget_registered_actions3sz Callbacks.get_registered_actionsF)threadcOsf||jvs Jd|d|j|j|D]6}|rNtj|d||ddq*|d|i|q*dS)aX Loop through the registered actions and fire all callbacks on main thread Args: hook: The name of the hook to check, defaults to all args: Arguments to receive from YOLOv5 thread: (boolean) Run callbacks in daemon thread kwargs: Keyword Arguments to receive from YOLOv5 rrrT)targetargskwargsdaemonN)r threadingThreadstart)rr r#r%r&loggerrrrrun<s z Callbacks.run)rN)N)__name__ __module__ __qualname____doc__rr!r"r,rrrrr s  r)r0r(rrrrrs