a Df @sddlZddlZddlmZz ddlZWneefyBdZYn0ddlZee Z dZ e ddZ e ddZdd Zd d Zd d ZddZdS)N)urlparsezcomet://COMET_MODEL_NAMEyolov5!COMET_DEFAULT_CHECKPOINT_FILENAMEzlast.ptc Csl|jd|j}tj|ddt}||}t|dkrNtd|dSt |dddd}d d |D}t |j }|j }|r| |}n| t}t}|durtd |d dSzrtd ||} |j|ddd} |d| } t| d} | | Wdn1s0Y| |_ Wn:tyf} z tdt| WYd} ~ n d} ~ 00dS)N/Texist_okrz3COMET ERROR: No checkpoints found for model name : cSs|dS)Nstep)xr r U/data/NAS7/SABIOD/METHODE/ermites/yolo-dyni/yolov5/utils/loggers/comet/comet_utils.py z+download_model_checkpoint..)keyreversecSsi|]}|d|dqS)fileNameassetIdr ).0assetr r r #rz-download_model_checkpoint..zCOMET ERROR: Checkpoint z" not found in the given Experimentz#COMET INFO: Downloading checkpoint binaryF return_typestreamwbz7COMET WARNING: Unable to download checkpoint from Comet)projectnameosmakedirsrZget_model_asset_listlenloggererrorsortedrweightsquerygetrinfo get_assetopenwrite Exceptionwarning exception)opt experiment model_dir model_nameZmodel_asset_listZlogged_checkpoint_mapZ resource_urlZcheckpoint_filenameasset_idZasset_filenameZ model_binaryZmodel_download_pathfer r r download_model_checkpointsB      *  r4c Cs|}|j}|D]V}|ddkr|d}|j|ddd}t|}|D]\}} t||| qL||_q|jd|j} t j | dd | d } t | d } t |j | Wd n1s0Y| |_ d S) zUpdate the opts Namespace with parameters from Comet's ExistingExperiment when resuming a run Args: opt (argparse.Namespace): Namespace of command line options experiment (comet_ml.APIExperiment): Comet API Experiment object rzopt.yamlrrFrrTrz /hyp.yamlwN)Zget_asset_listresumer'yaml safe_loaditemssetattrrrrrr(dumphyp) r-r.Z asset_listZ resume_stringrr1Z asset_binaryopt_dictrvaluesave_dirZ hyp_yaml_pathr2r r r set_opt_parametersBs     ,r@cCsbtdur dSt|jtr^|jtr^t}t|j}|j|j }| |}t ||dSdS)aMDownloads model weights from Comet and updates the weights path to point to saved weights location Args: opt (argparse.Namespace): Command Line arguments passed to YOLOv5 training script Returns: None/bool: Return True if weights are successfully downloaded else return None NT) comet_ml isinstancer#str startswith COMET_PREFIXAPIrnetlocpathr%r4r-apiresourceZexperiment_pathr.r r r check_comet_weightsas      rLcCsltdur dSt|jtrh|jtrht}t|j}|j|j }| |}t ||t ||dSdS)aURestores run parameters to its original state based on the model checkpoint and logged Experiment parameters. Args: opt (argparse.Namespace): Command Line arguments passed to YOLOv5 training script Returns: None/bool: Return True if the run is restored successfully else return None NT) rArBr6rCrDrErFrrGrHr%r@r4rIr r r check_comet_resume|s       rM)loggingr urllib.parserrAModuleNotFoundError ImportErrorr7 getLogger__name__r rEgetenvrrr4r@rLrMr r r r s      /