|
| list | action_demo_with_video.sys_path = ['/workspace/smlab', '/workspace'] |
| |
| str | action_demo_with_video.det_cfg_path = '/workspace/smlab/smdetect/configs/yolo/yolov8.py' |
| |
| str | action_demo_with_video.track_cfg_path = '/workspace/smlab/smtrack/configs/bytetrack/bytetrack_base.py' |
| |
| str | action_demo_with_video.pose_cfg_path = '/workspace/smlab/smpose/configs/mmpose/hrnet_trt.py' |
| |
| str | action_demo_with_video.action_cfg_path = '/workspace/InnoTest/models/posec3d_action.py' |
| |
| str | action_demo_with_video.posture_cfg_path = '/workspace/InnoTest/models/posec3d_pose.py' |
| |
| str | action_demo_with_video.device = 'cuda:0' |
| |
| | action_demo_with_video.det_model = smrunner.build_model(det_cfg_path) |
| |
| | action_demo_with_video.track_model = smrunner.build_model(track_cfg_path) |
| |
| | action_demo_with_video.pose_model = smrunner.build_model(pose_cfg_path) |
| |
| | action_demo_with_video.action_model = smrunner.build_model(action_cfg_path).to(device) |
| |
| | action_demo_with_video.posture_model = smrunner.build_model(posture_cfg_path).to(device) |
| |
| int | action_demo_with_video.action_sample = 20 |
| |
| int | action_demo_with_video.pose_sample = 6 |
| |
| | action_demo_with_video.labelmap = dict() |
| |
| list | action_demo_with_video.cvt_lower_map = [0, 1, 2, 0, 4, 5, 0, 7, 8, 9, 0, 11] |
| |
| list | action_demo_with_video.cvt_upper_map = [0, 0, 0, 3, 0, 0] |
| |
| list | action_demo_with_video.cvt_pose_map = [0, 0, 0, 0, 4, 4, 6, 0] |
| |
| list | action_demo_with_video.cvt_hand_map = [0, 0, 0, 3] |
| |
| | action_demo_with_video.action_vis_param |
| |
| | action_demo_with_video.vis_param |
| |
| str | action_demo_with_video.video_folder = '/media/safemotion/HDD5/pjm_test/ai_park_test_video' |
| |
| | action_demo_with_video._ |
| |
| | action_demo_with_video.video_path_list |
| |
| str | action_demo_with_video.save_folder = '/media/safemotion/HDD5/pjm_test/ai_park_test_fps20' |
| |
| str | action_demo_with_video.tmp_save_folder = '/media/safemotion/HDD5/pjm_test/tmp_clip_images_fps20' |
| |
| | action_demo_with_video.video_name = video_path.split('/')[-1] |
| |
| | action_demo_with_video.save_video_path = os.path.join(save_folder, video_name) |
| |
| | action_demo_with_video.cap = cv2.VideoCapture(video_path) |
| |
| | action_demo_with_video.total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) |
| |
| | action_demo_with_video.fps = cap.get(cv2.CAP_PROP_FPS) |
| |
| int | action_demo_with_video.action_k = int(fps+0.5)*2 |
| |
| int | action_demo_with_video.pose_k = int(fps+0.5)//2 |
| |
| dict | action_demo_with_video.track_data = {} |
| |
| int | action_demo_with_video.frame_id = 0 |
| |
| | action_demo_with_video.ret |
| |
| | action_demo_with_video.image |
| |
| | action_demo_with_video.end |
| |
| | action_demo_with_video.img_shape = image.shape[:2] |
| |
| | action_demo_with_video.vis_img = image.copy() |
| |
| | action_demo_with_video.det_result = det_model.run_detector(image) |
| |
| | action_demo_with_video.track_result = track_model.run_tracker(det_result['det_bboxes'], det_result['det_labels'], frame_id) |
| |
| | action_demo_with_video.pose_result = pose_model.run_detector(image, track_result['track_bboxes'][0]) |
| |
| | action_demo_with_video.track_id = int(pose['track_id']) |
| |
| list | action_demo_with_video.delete_list = [] |
| |
| int | action_demo_with_video.txt_pos_y = 50 |
| |
| | action_demo_with_video.bbox = pose_q[-1]['bbox'] |
| |
| | action_demo_with_video.last_frame = pose_q[-1]['frame_id'] |
| |
| str | action_demo_with_video.label = f'{track_id:3d}' |
| |
| | action_demo_with_video.action_result = action_model.run_recognizer(pose_q, action_k, action_sample, device=device) |
| |
| | action_demo_with_video.posture_result = posture_model.run_recognizer(pose_q[-pose_k:], pose_k, pose_sample, False, device=device) |
| |
| list | action_demo_with_video.pose_label = cvt_pose_map[posture_result['pred_pose']] |
| |
| list | action_demo_with_video.upper_label = cvt_upper_map[action_result['pred_action_upper']] |
| |
| list | action_demo_with_video.lower_label = cvt_lower_map[action_result['pred_action_lower']] |
| |
| list | action_demo_with_video.hand_label = cvt_hand_map[posture_result['pred_hand']] |
| |
| str | action_demo_with_video.name = f'{frame_id:09d}.jpg' |
| |
| | action_demo_with_video.save_path = os.path.join(tmp_save_folder, name) |
| |
| | action_demo_with_video.half |
| |
