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)