批量处理NTU数据集

'''
读取NTU原始数据的x y z 坐标，
保存在字典中
并将文件保存
'''
import os
import numpy as np
import pickle


def get_raw_body_data(ske_file,ske_name):


     assert os.path.exists(ske_file),'Error:Skeleton file {} not fond'.format(ske_file)
     print("Reading data from {}".format(ske_file))
     with open(ske_file,'r') as fr :
          str_data = fr.readlines()
     num_frames = int(str_data[0].strip('\r\n'))

     print("此文件共：{}帧".format(num_frames))
     frame_drop = []
     bodies_data = dict()
     valid_frames = -1
     current_line = 1

     for f in range(num_frames):
          num_bodies = int(str_data[current_line].strip('\r\n'))
          current_line += 1
          print("正在读取第{}帧".format(f))
          if num_frames == 0 :
               frame_drop.append(f)
               continue

          valid_frames += 1
          joints = np.zero