Reference for ultralytics/models/yolo/pose/predict.py
Note
This file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/pose/predict.py. If you spot a problem please help fix it by contributing a Pull Request 🛠️. Thank you 🙏!
ultralytics.models.yolo.pose.predict.PosePredictor
Bases: DetectionPredictor
A class extending the DetectionPredictor class for prediction based on a pose model.
Example
Source code in ultralytics/models/yolo/pose/predict.py
__init__(cfg=DEFAULT_CFG, overrides=None, _callbacks=None)
Initializes PosePredictor, sets task to 'pose' and logs a warning for using 'mps' as device.
Source code in ultralytics/models/yolo/pose/predict.py
postprocess(preds, img, orig_imgs)
Return detection results for a given input image or list of images.