Skip to content

Commit

Permalink
new approach: do not call mmdet bbox_thr. It's broken. Manually check…
Browse files Browse the repository at this point in the history
…ing confs instead.
  • Loading branch information
Cameron Johnson committed Jun 12, 2024
1 parent d38638b commit be6df9b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tcn_hpl/data/utils/pose_generation/rt_pose_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,14 +81,14 @@ def predict_single(det_model, pose_model, image: torch.tensor, bbox_thr=None) ->
current_ann["label"] = pred_label
current_ann["bbox_score"] = f"{scores[box_id] * 100:0.2f}"

if box_class == 0:
if box_class == 0 and float(current_ann["bbox_score"]) > bbox_thr:
person_results = [current_ann]

pose_results, returned_outputs = inference_top_down_pose_model(
model=pose_model,
img_or_path=image,
person_results=person_results,
bbox_thr=bbox_thr,
bbox_thr=None,
format="xyxy",
dataset=pose_dataset,
dataset_info=pose_dataset_info,
Expand Down

0 comments on commit be6df9b

Please sign in to comment.