device_config = k4a.K4A_DEVICE_CONFIG_INIT_DISABLE_ALL device_config.depth_mode = k4a.K4A_DEPTH_MODE_NFOV_UNBINNED # open file playback_handle = k4a.k4a_playback_t() mkv_path = 'G:/dataset/installing_1.mkv' VERIFY( k4a.k4a_playback_open( ctypes.c_char_p(bytes(mkv_path, encoding='utf8')), ctypes.byref(playback_handle)), "Cannot open recording {}!".format(mkv_path)) sensor_calibration = k4a.k4a_calibration_t() VERIFY( k4a.k4a_playback_get_calibration(playback_handle, ctypes.byref(sensor_calibration)), "Get depth camera calibration failed!") dep_in_cx = sensor_calibration.depth_camera_calibration.intrinsics.parameters.param.cx dep_in_cy = sensor_calibration.depth_camera_calibration.intrinsics.parameters.param.cy dep_in_fx = sensor_calibration.depth_camera_calibration.intrinsics.parameters.param.fx dep_in_fy = sensor_calibration.depth_camera_calibration.intrinsics.parameters.param.fy dep_in_mat = np.mat([[dep_in_fx, 0, dep_in_cx], [0, dep_in_fy, dep_in_cy], [0, 0, 1]]) col_in_cx = sensor_calibration.color_camera_calibration.intrinsics.parameters.param.cx col_in_cy = sensor_calibration.color_camera_calibration.intrinsics.parameters.param.cy col_in_fx = sensor_calibration.color_camera_calibration.intrinsics.parameters.param.fx col_in_fy = sensor_calibration.color_camera_calibration.intrinsics.parameters.param.fy col_in_mat = np.mat([[col_in_fx, 0, col_in_cx], [0, col_in_fy, col_in_cy], [0, 0, 1]])
playback_handle = k4a.k4a_playback_t() for v_i, video_dir in enumerate(video_list): print_toolbar(v_i * 1.0 / len(video_list), '({:>5}/{:<5}) Processing data: '.format( v_i + 1, len(video_list))) video_name = video_dir.split('/')[-1].split('.')[0] label_name = video_dir.split('/')[-1].split('_')[0] # open file VERIFY(k4a.k4a_playback_open(ctypes.c_char_p(bytes(video_dir, encoding='utf8')), ctypes.byref(playback_handle)), "Cannot open recording {}!".format(video_dir)) sensor_calibration = k4a.k4a_calibration_t() VERIFY(k4a.k4a_playback_get_calibration(playback_handle, ctypes.byref(sensor_calibration)), "Get depth camera calibration failed!") tracker = k4a.k4abt_tracker_t() tracker_config = k4a.K4ABT_TRACKER_CONFIG_DEFAULT VERIFY(k4a.k4abt_tracker_create(ctypes.byref(sensor_calibration), tracker_config, ctypes.byref(tracker)), "Body tracker initialization failed!") js_dict = {'data':[], 'label': label_name,'label_index':label_index_map[label_name]} if video_name not in js_label_dict.keys(): js_label_dict[video_name] = {"has_skeleton": True, "label": label_name, "label_index": label_index_map[label_name]} frame_count = 0 while frame_count < 300: sensor_capture = k4a.k4a_capture_t() stream_result = k4a.k4a_playback_get_next_capture(playback_handle,ctypes.byref(sensor_capture)) if stream_result == k4a.K4A_STREAM_RESULT_EOF: