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 f53721d commit 4a10b7e
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions test/pose/test_dwpose.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,12 @@ def test_dwpose_estimate_2girls(self, img_file_2girls, pose_data_2girls_0, pose_
assert len(poses) == 2
assert isinstance(poses[0], OP18KeyPointSet)
assert isinstance(poses[1], OP18KeyPointSet)
assert np.isclose(pose_data_2girls_0, poses[0].all, atol=1).all()
assert np.isclose(pose_data_2girls_1, poses[1].all, atol=1).all()
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}.'
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}.'

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

0 comments on commit 4a10b7e

Please sign in to comment.