def __init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, data_dirpath, processor_config, **kwargs): Processor.__init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, **kwargs) self.data_dirpath = data_dirpath self.fps = processor_config['fps'] self.kinect_dump_fp = processor_config.get('dump_fp', None) self.realtime = self.kinect_dump_fp is None self.skip_old_atol_ms = 0.2 \ if processor_config.get('skip_old', False) else None self.get_color_timestamp = processor_config.get( 'get_color_timestamp', True) self.parallel_bt = processor_config.get('parallel_bt', True) self.log_level = processor_config.get('log_level', 2) self.gpu_id = processor_config.get('gpu_id', 0) self.face_cropper = None self.prev_time = time.time() self.k4a = None self.kinect_dump_frames = None self.decoded_cam_params = None self.index = -1 self.dump_index = -1 self.output_count = 0
def __init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, sleep_args, use_kinect): Processor.__init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue) self.sleeper = Sleeper(*sleep_args) self.index = -1 self.use_kinect = use_kinect if self.use_kinect: self.k4a = None
def __init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, data_dirpath, processor_config, **kwargs): Processor.__init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, **kwargs) self.data_dirpath = data_dirpath self.processor_config = processor_config self.log_level = processor_config.get('log_level', 2) self.model = None self.flip_side = 'right' self.index = -1 self.output_count = 0
def __init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, data_dirpath, processor_config, **kwargs): Processor.__init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, **kwargs) self.data_dirpath = data_dirpath self.processor_config = processor_config self.log_level = processor_config.get('log_level', 2) self.device = self.processor_config['device'] self.inferer = None self.index = -1 self.output_count = 0
def __init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, data_dirpath, processor_config, pykinect_data_dp, output_dirpath, **kwargs): Processor.__init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, **kwargs) self.data_dirpath = data_dirpath self.processor_config = processor_config self.log_level = processor_config.get('log_level', 2) self.pykinect_data_dp = pykinect_data_dp self.filterer_config = self.processor_config['filterer'] self.filterer_config['body_models_dp'] = osp.join( self.pykinect_data_dp, 'body_models') self.output_dirpath = output_dirpath self.gender = self.processor_config['gender'] self.beta = np.load( osp.join(self.data_dirpath, self.processor_config['person_shape_path'], 'betas.npy')) self.vis_config = self.processor_config['vis_pose'] self.smplx_wrapper = None self.visualizer = None if self.vis_config['enable'] and self.vis_config['imsave']: self.vis_dp = osp.join(self.output_dirpath, 'aggregate') os.makedirs(self.vis_dp, exist_ok=True) else: self.vis_dp = None self.filterer = None self.decoded_cam_params = None self.smplx_device = None self.visualize_device = None self.prev_time = time.time() self.dump_config = self.processor_config.get('dump', None) if self.dump_config is not None and self.dump_config['enable']: self.dump_dp = osp.join(self.output_dirpath, self.dump_config['output_path']) os.makedirs(self.dump_dp, exist_ok=True) self.dump_fp = osp.join(self.dump_dp, 'poses.pickle') self.dump_data = [] else: self.dump_data = None self.index = -1 self.output_count = 0
def __init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, data_dirpath, processor_config, pykinect_data_dp, **kwargs): Processor.__init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, **kwargs) self.data_dirpath = data_dirpath self.processor_config = processor_config self.pykinect_data_dp = pykinect_data_dp self.log_level = processor_config.get('log_level', 2) self.device = torch.device(self.processor_config['device']) self.gender = self.processor_config['gender'] self.beta = np.load( osp.join(self.data_dirpath, self.processor_config['person_shape_path'], 'betas.npy')) self.model_dp = osp.join(self.data_dirpath, self.processor_config['model_path']) self.inferer = None self.index = -1 self.output_count = 0
def __init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue, sleep_args): Processor.__init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue) self.sleeper = Sleeper(*sleep_args)
def __init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue): Processor.__init__(self, name, msg_queue, input_queue, output_queue, assembler_input_queue)