From 29a93a3c6865863b27461721d72b6153ea5484cf Mon Sep 17 00:00:00 2001 From: iabdalkader Date: Sat, 7 Jan 2017 18:53:05 +0200 Subject: [PATCH] Update keypoints editor to draw angles. --- util/keypoints_editor.py | 33 +++++++++++++++++++++------------ 1 file changed, 21 insertions(+), 12 deletions(-) diff --git a/util/keypoints_editor.py b/util/keypoints_editor.py index 8dae1fffb..3c21a99fb 100755 --- a/util/keypoints_editor.py +++ b/util/keypoints_editor.py @@ -4,19 +4,21 @@ import time import struct import pygame import sys +from math import sin, cos, radians -SCALE = 2 +SCALE = 3 +RED = (255,0,0) GREEN = (0,255,0) GRAY = (127, 127, 127) -KEYPOINTS_SIZE = 16 +KEYPOINTS_SIZE = 24 def load_keypoints(path): kpts = [] with open(path, "rb") as f: size = struct.unpack("