{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "# Preprocessing of squat or pushup raw coordinate text files into segmented and normalized reps" ] }, { "cell_type": "code", "execution_count": 1, "metadata": { "collapsed": false }, "outputs": [ { "name": "stderr", "output_type": "stream", "text": [ "/Users/Brandon/anaconda/lib/python2.7/site-packages/matplotlib/font_manager.py:273: UserWarning: Matplotlib is building the font cache using fc-list. This may take a moment.\n", " warnings.warn('Matplotlib is building the font cache using fc-list. This may take a moment.')\n" ] } ], "source": [ "#=====[ Import dependencies ]====\n", "import numpy as np\n", "import matplotlib.pyplot as plt\n", "import sys\n", "# sys.path.append('/usr/local/lib/python2.7/site-packages')\n", "import pandas as pd\n", "import pickle\n", "\n", "# sys.path.append('/Users/admin3/Dropbox/Stanford/Fall-2015/CS221/project/ai_trainer/data')\n", "sys.path.append('../data')\n", "sys.path.append('../data/squat_pushupData_10to20')\n", "sys.path.append('../inference')\n", "\n", "#=====[ Import Data ]=====\n", "# import coords1 as cd\n", "import coordKeys as keys\n", "import coordKeysZ as keysZ\n", "\n", "#=====[ Import Labels ]======\n", "import labels\n", "\n", "%matplotlib inline" ] }, { "cell_type": "markdown", "metadata": { "collapsed": true }, "source": [ "# Step 1: Import Data" ] }, { "cell_type": "code", "execution_count": 5, "metadata": { "collapsed": false }, "outputs": [], "source": [ "#=====[ Get labels ]=====\n", "y = labels.labels15\n", "# print 'Number of labeled squats: %d' % len(y)\n", "\n", "#=====[ Get raw text files that specify x, y, z coordinates for each joint for each frame ]=====\n", "file_to_open = '../data/raw_data/squat_pushupData_10to20/squatData13.txt'\n", "# file_to_save = '../data/squat_coords/squat17.p'" ] }, { "cell_type": "code", "execution_count": 6, "metadata": { "collapsed": false }, "outputs": [ { "data": { "text/html": [ "
\n", " | SpineBaseX | \n", "SpineBaseY | \n", "SpineBaseZ | \n", "SpineMidX | \n", "SpineMidY | \n", "SpineMidZ | \n", "NeckX | \n", "NeckY | \n", "NeckZ | \n", "HeadX | \n", "... | \n", "HandTipLeftZ | \n", "ThumbLeftX | \n", "ThumbLeftY | \n", "ThumbLeftZ | \n", "HandTipRightX | \n", "HandTipRightY | \n", "HandTipRightZ | \n", "ThumbRightX | \n", "ThumbRightY | \n", "ThumbRightZ | \n", "
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | \n", "263.104919 | \n", "174.818909 | \n", "2.459708 | \n", "264.268982 | \n", "125.396904 | \n", "2.454690 | \n", "265.416779 | \n", "76.684052 | \n", "2.437456 | \n", "265.499786 | \n", "... | \n", "2.305079 | \n", "243.933044 | \n", "192.064728 | \n", "2.268467 | \n", "383.498108 | \n", "50.996933 | \n", "2.374507 | \n", "368.429321 | \n", "58.428131 | \n", "2.406143 | \n", "
1 | \n", "263.215820 | \n", "175.213943 | \n", "2.414441 | \n", "264.262238 | \n", "125.240326 | \n", "2.442008 | \n", "265.403656 | \n", "76.679108 | \n", "2.437677 | \n", "265.522247 | \n", "... | \n", "2.426420 | \n", "164.144028 | \n", "44.143250 | \n", "2.450143 | \n", "377.598572 | \n", "43.199921 | \n", "2.358433 | \n", "364.713623 | \n", "49.998352 | \n", "2.416857 | \n", "
2 | \n", "264.136200 | \n", "173.187103 | \n", "2.412615 | \n", "264.593689 | \n", "124.594658 | \n", "2.437691 | \n", "265.411591 | \n", "76.689972 | \n", "2.437383 | \n", "265.539978 | \n", "... | \n", "2.429082 | \n", "173.382568 | \n", "37.293640 | \n", "2.426909 | \n", "370.047913 | \n", "40.323441 | \n", "2.370817 | \n", "358.668152 | \n", "45.335098 | \n", "2.417667 | \n", "
3 | \n", "264.404480 | \n", "172.264862 | \n", "2.411652 | \n", "264.741241 | \n", "124.129387 | \n", "2.435206 | \n", "265.411469 | \n", "76.691391 | \n", "2.437093 | \n", "265.548248 | \n", "... | \n", "2.431059 | \n", "184.625061 | \n", "40.621124 | \n", "2.442875 | \n", "362.123566 | \n", "33.202423 | \n", "2.379206 | \n", "350.555969 | \n", "40.885727 | \n", "2.414000 | \n", "
4 | \n", "264.476807 | \n", "172.044189 | \n", "2.411136 | \n", "264.880463 | \n", "123.929939 | \n", "2.433811 | \n", "265.483307 | \n", "76.702988 | \n", "2.436774 | \n", "265.549957 | \n", "... | \n", "2.411451 | \n", "192.250656 | \n", "35.749924 | \n", "2.449250 | \n", "352.409241 | \n", "25.838058 | \n", "2.391559 | \n", "342.909454 | \n", "38.182144 | \n", "2.420182 | \n", "
5 rows × 75 columns
\n", "