o H!i@s,ddlmZddlmZGdddeZdS)) DetTrainer)MODELSc@seZdZdZeZddZdS)KeypointTrainerz#Human Pose Estimation Model TrainercCs@|jj}d|jjvr|jjdnd}|jj|jjd||ddS)zupdate dataset settings data_fieldsNZKeypointTopDownCocoDataset)rmetric)Z pdx_configrZ TrainDatasetZupdate_datasetZ global_configZ dataset_dir)selfrrr v/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/paddlex/modules/keypoint_detection/trainer.py_update_datasets   zKeypointTrainer._update_datasetN)__name__ __module__ __qualname____doc__rentitiesr r r r r rs rN)Zobject_detectionrZ model_listrrr r r r s