Skip to content

Commit fcf53af

Browse files
authored
Merge pull request #35 from unnonouno/fix-matrix
Use list instead of str for initializer of matrix object
2 parents 0a10913 + 4366095 commit fcf53af

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,7 @@ def apply_rotate(self, image, joints, ignore_joints):
129129
image = (image * 255).astype(np.uint8)
130130
theta = -np.radians(angle)
131131
c, s = np.cos(theta), np.sin(theta)
132-
rot_mat = np.matrix('{} {}; {} {}'.format(c, -s, s, c))
132+
rot_mat = np.matrix([[c, -s], [s, c]])
133133
joints = rot_mat.dot((joints - joint_center).T).T + joint_center
134134
return image, np.array(joints.tolist())
135135

0 commit comments

Comments
 (0)