Fig BddlmZddlmZddlmZmZGddeZy)) 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() NcJt||||d|j_y)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 overrides _callbacks __class__s f/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/models/yolo/pose/predict.pyr zPosePredictor.__init__s! i4 c&t|||||}|ddddfj|jdg|jj }t j|jdd||j}|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")$XxH$DABK$$TZZ]JTZZ5I5IJ $$SYYqr]Ix~~N  * r)rz dict | None)__name__ __module__ __qualname____doc__rr r __classcell__)rs@rrr s!('$RV  rrN) __future__r&ultralytics.models.yolo.detect.predictrultralytics.utilsrrrrrr0s#E.:&:r