generate_bbox

lightning_pose.utils.cropzoom.generate_bbox(input_preds_file: Path, detector_cfg: DictConfig, output_bbox_file: Path) → None[source]

Compute bounding boxes from model predictions and save to a CSV file.

Parameters:

  • input_preds_file – path to the predictions CSV produced by litpose predict.

  • detector_cfg – OmegaConf config containing anchor_keypoints and either crop_ratio or crop_height/crop_width.

  • output_bbox_file – path where the bbox CSV will be written.