cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make image preprocessing support RGBA image.

Browse Source 
PiperOrigin-RevId: 513680594
This commit is contained in:
MediaPipe Team 2023-03-02 17:11:04 -08:00 committed by Copybara-Service
parent c3adf19706
commit 91d53cd181
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/tasks/cc/vision/utils/image_tensor_specs.cc
View File

@ -182,11 +182,11 @@ absl::StatusOr<ImageTensorSpecs> BuildInputImageTensorSpecs(
"Only RGB color space is supported for now.", "Only RGB color space is supported for now.",
MediaPipeTasksStatus::kInvalidArgumentError); MediaPipeTasksStatus::kInvalidArgumentError);
} }
if (batch != 1 || depth != 3) { if (batch != 1 || (depth != 3 && depth != 4)) {
return CreateStatusWithPayload( return CreateStatusWithPayload(
StatusCode::kInvalidArgument, StatusCode::kInvalidArgument,
absl::StrCat("The input tensor should have dimensions 1 x height x " absl::StrCat("The input tensor should have dimensions 1 x height x "
"width x 3. Got ", "width x depth, where depth = 3 or 4. Got ",
batch, " x ", height, " x ", width, " x ", depth, "."), batch, " x ", height, " x ", width, " x ", depth, "."),
MediaPipeTasksStatus::kInvalidInputTensorDimensionsError); MediaPipeTasksStatus::kInvalidInputTensorDimensionsError);
} }