adding data+code to git repo

1 files changed, 36 insertions, 0 deletions

diff --git a/data/segment.py b/data/segment.py
new file mode 100755
index 0000000..a2cfdb7
--- /dev/null
+++ b/data/segment.py
@@ -0,0 +1,36 @@
+#!/usr/bin/python
+
+import time
+import sys
+import math
+import copy
+
+frames = [0,0,0,0,0,0,0,0,0,0]
+pframes = [0,0,0,0,0,0,0,0,0,0]
+pframe = 0
+lines = [[],[],[],[],[],[],[],[],[],[]]
+num = 1
+
+for arg in sys.argv[1:]:
+    sk_file = open(arg)
+    for line in sk_file:
+        frame, id, joint, state, x, y, z, dx, dy = line.strip().split(',')
+        if frame != 'Frame':
+            id = int(id)
+            frame = int(frame)
+            if frame != pframe:
+                for i in range(len(frames)):
+                    if frames[i] > 0 and pframe > pframes[i]+1:
+                        #print str(frames[i])
+			if frames[i] >= 50 and frames[i] <= 100:
+				for l in lines[i]:
+					print str(num)+','+l
+				num += 1
+                        frames[i] = 0
+			lines[i] = []
+                pframe = frame
+            if pframes[id] != frame:
+                frames[id] += 1
+                pframes[id] = frame 
+	    lines[id] += [line.strip()]
+