/jg JddlmZddlmZddlmZmZGddeZdS)) annotations)DetectionPredictor) DEFAULT_CFGopsc6eZdZdZeddfdfd ZfdZxZS) PosePredictoraAA class extending the DetectionPredictor class for prediction based on a pose model. This class specializes in pose estimation, handling keypoints detection alongside standard object detection capabilities inherited from DetectionPredictor. Attributes: args (namespace): Configuration arguments for the predictor. model (torch.nn.Module): The loaded YOLO pose model with keypoint detection capabilities. Methods: construct_result: Construct the result object from the prediction, including keypoints. Examples: >>> from ultralytics.utils import ASSETS >>> from ultralytics.models.yolo.pose import PosePredictor >>> args = dict(model="yolo26n-pose.pt", source=ASSETS) >>> predictor = PosePredictor(overrides=args) >>> predictor.predict_cli() N _callbacks dict | Nonecft|||d|j_dS)aInitialize PosePredictor for pose estimation tasks. Sets up a PosePredictor instance, configuring it for pose detection tasks and handling device-specific warnings for Apple MPS. Args: cfg (Any): Configuration for the predictor. overrides (dict, optional): Configuration overrides that take precedence over cfg. _callbacks (dict, optional): Dictionary of callback functions to be invoked during prediction. poseN)super__init__argstask)selfcfg overridesr __class__s i/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/models/yolo/pose/predict.pyrzPosePredictor.__init__s. i444 c*t||||}|ddddfj|jdg|jjR}t j|jdd||j}|||S)aYConstruct the result object from the prediction, including keypoints. Extends the parent class implementation by extracting keypoint data from predictions and adding them to the result object. Args: pred (torch.Tensor): The predicted bounding boxes, scores, and keypoints with shape (N, 6+K*D) where N is the number of detections, K is the number of keypoints, and D is the keypoint dimension. img (torch.Tensor): The processed input image tensor with shape (B, C, H, W). orig_img (np.ndarray): The original unprocessed image as a numpy array. img_path (str): The path to the original image file. Returns: (Results): The result object containing the original image, image path, class names, bounding boxes, and keypoints. Nr) keypoints) r construct_resultviewshapemodel kpt_shaper scale_coordsupdate)rpredimgorig_imgimg_pathresult pred_kptsrs rrzPosePredictor.construct_result,s"))$XxHH$DABBK$TZ]JTZ5IJJJ $SYqrr]Ix~NN  *** r)r r )__name__ __module__ __qualname____doc__rrr __classcell__)rs@rrr so('$RV        rrN) __future__r&ultralytics.models.yolo.detect.predictrultralytics.utilsrrrrrr1sx#"""""EEEEEE........:::::&:::::r