o W h@s4ddlmZddlmZmZmZGdddeZdS))DetectionPredictor) DEFAULT_CFGLOGGERopscs4eZdZdZeddffdd ZfddZZS) PosePredictoraG A 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: Constructs the result object from the prediction, including keypoints. Examples: >>> from ultralytics.utils import ASSETS >>> from ultralytics.models.yolo.pose import PosePredictor >>> args = dict(model="yolo11n-pose.pt", source=ASSETS) >>> predictor = PosePredictor(overrides=args) >>> predictor.predict_cli() NcsLt|||d|j_t|jjtr"|jjdkr$t ddSdSdS)aW Initialize PosePredictor, a specialized predictor for pose estimation tasks. This initializer 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. Default is DEFAULT_CFG. overrides (dict, optional): Configuration overrides that take precedence over cfg. _callbacks (list, optional): List of callback functions to be invoked during prediction. Examples: >>> from ultralytics.utils import ASSETS >>> from ultralytics.models.yolo.pose import PosePredictor >>> args = dict(model="yolov8n-pose.pt", source=ASSETS) >>> predictor = PosePredictor(overrides=args) >>> predictor.predict_cli() posempsz}Apple MPS known Pose bug. Recommend 'device=cpu' for Pose models. See https://github.com/ultralytics/ultralytics/issues/4031.N) super__init__argstask isinstancedevicestrlowerrwarning)selfcfg overrides _callbacks __class__X/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/models/yolo/pose/predict.pyr szPosePredictor.__init__cst||||}t|r"|ddddfjt|g|jjRn |ddddf}t|jdd||j}|j |d|S)a^ Construct the result object from the prediction, including keypoints. This method 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. N) keypoints) r construct_resultlenviewmodel kpt_shaper scale_coordsshapeupdate)rpredimgorig_imgimg_pathresult pred_kptsrrrr8s F zPosePredictor.construct_result)__name__ __module__ __qualname____doc__rr r __classcell__rrrrrsrN)&ultralytics.models.yolo.detect.predictrultralytics.utilsrrrrrrrrs