Skip to content

Commit 292e38c

Browse files
Fixed issue with saved pose images (#1972) (#1973)
Co-authored-by: Eugene Khvedchenya <ekhvedchenya@gmail.com>
1 parent 3548fd5 commit 292e38c

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/super_gradients/training/utils/predict/prediction_pose_estimation_results.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,14 @@ def save(
119119
:param show_confidence: Whether to show confidence scores on the image.
120120
:param box_thickness: (Optional) Thickness of bounding boxes. If None, will adapt to the box size.
121121
"""
122-
image = self.draw(box_thickness=box_thickness, show_confidence=show_confidence)
122+
image = self.draw(
123+
edge_colors=edge_colors,
124+
joint_thickness=joint_thickness,
125+
keypoint_colors=keypoint_colors,
126+
keypoint_radius=keypoint_radius,
127+
box_thickness=box_thickness,
128+
show_confidence=show_confidence,
129+
)
123130
save_image(image=image, path=output_path)
124131

125132

0 commit comments

Comments
 (0)