Skip to content

Commit

Permalink
dev(narugo): fix the dwpose unittest
Browse files Browse the repository at this point in the history
  • Loading branch information
narugo1992 committed Dec 8, 2024
1 parent 4a10b7e commit 5711095
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/pose/test_dwpose.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,10 +31,10 @@ def test_dwpose_estimate_2girls(self, img_file_2girls, pose_data_2girls_0, pose_
assert isinstance(poses[1], OP18KeyPointSet)
c0 = np.isclose(pose_data_2girls_0, poses[0].all, atol=1)
assert c0.all(), f'{c0.mean() * 100:.2f}% of the values are the same, ' \
f'max diff is {(pose_data_2girls_0 - poses[0].all).max():.4f}.'
f'max diff is {np.abs(pose_data_2girls_0 - poses[0].all).max():.4f}.'
c1 = np.isclose(pose_data_2girls_1, poses[1].all, atol=1)
assert c1.all(), f'{c1.mean() * 100:.2f}% of the values are the same, ' \
f'max diff is {(pose_data_2girls_1 - poses[1].all).max():.4f}.'
f'max diff is {np.abs(pose_data_2girls_1 - poses[1].all).max():.4f}.'

def test_dwpose_estimate_nothing(self, img_file_gun):
poses = dwpose_estimate(img_file_gun)
Expand Down

0 comments on commit 5711095

Please sign in to comment.