Skip to content

Commit

Permalink
Remove unnecessary processing
Browse files Browse the repository at this point in the history
  • Loading branch information
PINTO0309 committed Dec 15, 2024
1 parent d9f9081 commit 7e4940e
Showing 1 changed file with 0 additions and 10 deletions.
10 changes: 0 additions & 10 deletions 462_Gaze-LLE/demo/demo_yolov9_onnx_gazelle.py
Original file line number Diff line number Diff line change
Expand Up @@ -511,16 +511,6 @@ def _postprocess(
left_right_hand_boxes = [box for box in result_boxes if box.classid in [22, 23]]
self._find_most_relevant_obj(base_objs=hand_boxes, target_objs=left_right_hand_boxes)
result_boxes = [box for box in result_boxes if box.classid not in [22, 23]]

# Nose merge
# classid: 7 -> Head
# classid: 18 -> Nose
# 1. Calculate Nose IoUs for Head detection results
# 2. Connect either the Nose with the highest score and the highest IoU with the Head.
head_boxes = [box for box in result_boxes if box.classid == 7]
nose_boxes = [box for box in result_boxes if box.classid in [18]]
self._find_most_relevant_obj(base_objs=head_boxes, target_objs=nose_boxes)
result_boxes = [box for box in result_boxes if box.classid not in [18]]
return result_boxes

def _find_most_relevant_obj(
Expand Down

0 comments on commit 7e4940e

Please sign in to comment.