We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 63d564b commit 0a02dd1Copy full SHA for 0a02dd1
lib/utils/transforms.py
@@ -19,9 +19,9 @@ def flip_back(output_flipped, matched_parts):
19
assert output_flipped.ndim == 4,\
20
'output_flipped should be [batch_size, num_joints, height, width]'
21
22
- output_flipped = output_flipped[:, :, :, ::-1]
+ output_flipped = output_flipped[:, :, :, ::-1]#沿着width左右翻转
23
24
- for pair in matched_parts:
+ for pair in matched_parts:#大概率是用于tracking的吧 左右翻转之后关键点坐标也要翻一下??
25
tmp = output_flipped[:, pair[0], :, :].copy()
26
output_flipped[:, pair[0], :, :] = output_flipped[:, pair[1], :, :]
27
output_flipped[:, pair[1], :, :] = tmp
@@ -82,7 +82,7 @@ def get_affine_transform(
82
dst[2:, :] = get_3rd_point(dst[0, :], dst[1, :])
83
84
if inv:
85
- trans = cv2.getAffineTransform(np.float32(dst), np.float32(src))
+ trans = cv2.getAffineTransform(np.float32(dst), np.float32(src))#仿射变换 保持二维图形的平直性
86
else:
87
trans = cv2.getAffineTransform(np.float32(src), np.float32(dst))
88
0 commit comments