Fix depth condition bug when only depth condition is configured.

PiperOrigin-RevId: 567080598
This commit is contained in:
MediaPipe Team 2023-09-20 14:26:48 -07:00 committed by Copybara-Service
parent 41a012721f
commit 223641a73c

View File

@ -683,6 +683,7 @@ public final class ImageGenerator extends BaseVisionTaskApi {
.setEdgeConditionTypeOptions(edgeConditionOptions().get().convertToProto()) .setEdgeConditionTypeOptions(edgeConditionOptions().get().convertToProto())
.build()) .build())
.build()); .build());
}
if (depthConditionOptions().isPresent()) { if (depthConditionOptions().isPresent()) {
taskOptionsBuilder.addControlPluginGraphsOptions( taskOptionsBuilder.addControlPluginGraphsOptions(
ControlPluginGraphOptionsProto.ControlPluginGraphOptions.newBuilder() ControlPluginGraphOptionsProto.ControlPluginGraphOptions.newBuilder()
@ -696,7 +697,6 @@ public final class ImageGenerator extends BaseVisionTaskApi {
.build()) .build())
.build()); .build());
} }
}
return Any.newBuilder() return Any.newBuilder()
.setTypeUrl( .setTypeUrl(
"type.googleapis.com/mediapipe.tasks.vision.image_generator.proto.ImageGeneratorGraphOptions") "type.googleapis.com/mediapipe.tasks.vision.image_generator.proto.ImageGeneratorGraphOptions")