hanyuan.chy
3 years ago
1 changed files with 3 additions and 3 deletions
Unified View
Diff Options
+ 3
- 3
modelscope/pipelines/cv/body_3d_keypoints_pipeline.py
View File
| @@ -132,8 +132,8 @@ class Body3DKeypointsPipeline(Pipeline): | |||||
| device='gpu' if torch.cuda.is_available() else 'cpu') | device='gpu' if torch.cuda.is_available() else 'cpu') | ||||
| def preprocess(self, input: Input) -> Dict[str, Any]: | def preprocess(self, input: Input) -> Dict[str, Any]: | ||||
| video_url = input | |||||
| video_frames = self.read_video_frames(video_url) | |||||
| self.video_url = input | |||||
| video_frames = self.read_video_frames(self.video_url) | |||||
| if 0 == len(video_frames): | if 0 == len(video_frames): | ||||
| res = {'success': False, 'msg': 'get video frame failed.'} | res = {'success': False, 'msg': 'get video frame failed.'} | ||||
| return res | return res | ||||
| @@ -198,7 +198,7 @@ class Body3DKeypointsPipeline(Pipeline): | |||||
| } | } | ||||
| if not input['success']: | if not input['success']: | ||||
| pass | |||||
| res[OutputKeys.OUTPUT_VIDEO] = self.video_url | |||||
| else: | else: | ||||
| poses = input[KeypointsTypes.POSES_CAMERA] | poses = input[KeypointsTypes.POSES_CAMERA] | ||||
| pred_3d_pose = poses.data.cpu().numpy()[ | pred_3d_pose = poses.data.cpu().numpy()[ | ||||