Skip to content

Commit ace63af

Browse files
committed
update get_status_info: check valid_frames_mask is boolean array
1 parent f505883 commit ace63af

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

fmpose3d/inference_api/fmpose3d.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -645,6 +645,8 @@ def get_status_info(self) -> tuple[ResultStatus, str]:
645645
return ResultStatus.UNKNOWN, "No frame-validity mask provided by the 2D pose."
646646
if not isinstance(self.valid_frames_mask, np.ndarray) or self.valid_frames_mask.ndim != 1:
647647
return ResultStatus.UNKNOWN, "invalid 2D pose valid_frames_mask: must be a 1D numpy array."
648+
if not np.issubdtype(self.valid_frames_mask.dtype, np.bool_):
649+
return ResultStatus.UNKNOWN, "invalid 2D pose valid_frames_mask: must be a boolean numpy array."
648650
if self.valid_frames_mask.shape[0] != num_frames:
649651
return ResultStatus.INVALID, "2D pose valid_frames_mask mismatches the number of frames."
650652

@@ -698,6 +700,8 @@ def get_status_info(self) -> tuple[ResultStatus, str]:
698700
return ResultStatus.UNKNOWN, "No frame-validity mask provided by the 3D pose."
699701
if not isinstance(self.valid_frames_mask, np.ndarray) or self.valid_frames_mask.ndim != 1:
700702
return ResultStatus.UNKNOWN, "invalid 3D pose valid_frames_mask: must be a 1D numpy array."
703+
if not np.issubdtype(self.valid_frames_mask.dtype, np.bool_):
704+
return ResultStatus.UNKNOWN, "invalid 3D pose valid_frames_mask: must be a boolean numpy array."
701705
if self.valid_frames_mask.shape[0] != num_frames:
702706
return ResultStatus.INVALID, "3D pose valid_frames_mask mismatches the number of frames."
703707

0 commit comments

Comments
 (0)