Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 12:39:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix output video path when person detect failed.

Browse Source
This commit is contained in:
hanyuan.chy
2022-11-04 09:33:09 +00:00
parent da2de88b7a
commit 26db21d57d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
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')
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):
res = {'success': False, 'msg': 'get video frame failed.'}
return res
@@ -198,7 +198,7 @@ class Body3DKeypointsPipeline(Pipeline):
}
if not input['success']:
pass
res[OutputKeys.OUTPUT_VIDEO] = self.video_url
else:
poses = input[KeypointsTypes.POSES_CAMERA]
pred_3d_pose = poses.data.cpu().numpy()[