関数化（クラス化）の概念がわかりません

取得した骨格座標（17関節×2(x,y)）をcv2.lineを使って関節を描画したいです。右半身、左半身、中心部、右顔、左顔の５つで色分けして描画したいのですが、関数化などもなにも考えずに書いたら下のようなごちゃごちゃのコードになってしまいました。どうしたらもっとスッキリかけますか？

python
1def write_lines(img, key_points): 
2    key_points = key_points.reshape(-1, 2)
3
4    left_lines = [[3,5],[5,7],[7,9],[5,11],[11,13],[13,15]]  
5    right_lines = [[4,6],[6,8],[8,10],[6,12],[12,14],[14,16]]
6    middle_lines = [[5,6],[12,11]]
7    left_face_lines = [[0,1],[1,3]]
8    right_face_lines = [[0,2],[2,4]]
9
10    left_color = (0,0,255)
11    right_color = (255,0,0)
12    middle_color = (0,255,0)
13    left__face_color = (184,213,254)
14    right_face_color = (254,184,213)
15    
16#個々の部分がくどいような気がします。（ここを関数化できる？）
17    for start, end in left_lines:
18        cv2.line(img, (int(key_points[start, 0]), int(key_points[start, 1])), 
19                 (int(key_points[end, 0]), int(key_points[end, 1])), left_color, thickness=10)
20
21    for start, end in right_lines:
22        cv2.line(img, (int(key_points[start, 0]), int(key_points[start, 1])), 
23                 (int(key_points[end, 0]), int(key_points[end, 1])), right_color, thickness=10)
24
25    for start, end in middle_lines:
26        cv2.line(img, (int(key_points[start, 0]), int(key_points[start, 1])), 
27                 (int(key_points[end, 0]), int(key_points[end, 1])), middle_color, thickness=10)
28
29    for start, end in left_face_lines:
30        cv2.line(img, (int(key_points[start, 0]), int(key_points[start, 1])), 
31                 (int(key_points[end, 0]), int(key_points[end, 1])), left__face_color, thickness=5)
32
33    for start, end in right_face_lines:
34        cv2.line(img, (int(key_points[start, 0]), int(key_points[start, 1])), 
35                 (int(key_points[end, 0]), int(key_points[end, 1])), right_face_color, thickness=5)
36
37    return img

行動規範の内容に同意します

回答1件

ベストアンサー

こんな感じでしょうか？

python
1def write_lines(img, key_points):
2    key_points = key_points.reshape(-1, 2)
3
4    left_lines = (3, 5), (5, 7), (7, 9), (5, 11), (11, 13), (13, 15)
5    right_lines = (4, 6), (6, 8), (8, 10), (6, 12), (12, 14), (14, 16)
6    middle_lines = (5, 6), (12, 11)
7    left_face_lines = (0, 1), (1, 3)
8    right_face_lines = (0, 2), (2, 4)
9
10    left_color = (0, 0, 255)
11    right_color = (255, 0, 0)
12    middle_color = (0, 255, 0)
13    left__face_color = (184, 213, 254)
14    right_face_color = (254, 184, 213)
15
16    def draw(lines, color, thickness=10):
17        for start, end in lines:
18            cv2.line(img,
19                     (int(key_points[start, 0]), int(key_points[start, 1])),
20                     (int(key_points[end, 0]), int(key_points[end, 1])),
21                     color, thickness=thickness)
22
23    draw(left_lines, left_color)
24    draw(right_lines, right_color)
25    draw(middle_lines, middle_color)
26    draw(left_face_lines, left_face_color, 5)
27    draw(right_face_lines, right_left_color, 5)
28    return img
29