Reference for ultralytics/models/yolo/pose/train.py
Note
This file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/models/yolo/pose/train.py. If you spot a problem please help fix it by contributing a Pull Request 🛠️. Thank you 🙏!
ultralytics.models.yolo.pose.train.PoseTrainer
Bases: DetectionTrainer
A class extending the DetectionTrainer class for training based on a pose model.
Example
Source code in ultralytics/models/yolo/pose/train.py
get_model
Get pose estimation model with specified configuration and weights.
Source code in ultralytics/models/yolo/pose/train.py
get_validator
Returns an instance of the PoseValidator class for validation.
Source code in ultralytics/models/yolo/pose/train.py
plot_metrics
plot_training_samples
Plot a batch of training samples with annotated class labels, bounding boxes, and keypoints.
Source code in ultralytics/models/yolo/pose/train.py
set_model_attributes
📅 Created 12 months ago ✏️ Updated 1 month ago