Add output size as parameters in Java ImageSegmenter
PiperOrigin-RevId: 558834692
This commit is contained in:
parent
cd9d32e797
commit
737c103940
|
@ -237,6 +237,10 @@ public class PacketCreator {
|
||||||
return Packet.create(nativeCreateInt32Array(mediapipeGraph.getNativeHandle(), data));
|
return Packet.create(nativeCreateInt32Array(mediapipeGraph.getNativeHandle(), data));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public Packet createInt32Pair(int first, int second) {
|
||||||
|
return Packet.create(nativeCreateInt32Pair(mediapipeGraph.getNativeHandle(), first, second));
|
||||||
|
}
|
||||||
|
|
||||||
public Packet createFloat32Array(float[] data) {
|
public Packet createFloat32Array(float[] data) {
|
||||||
return Packet.create(nativeCreateFloat32Array(mediapipeGraph.getNativeHandle(), data));
|
return Packet.create(nativeCreateFloat32Array(mediapipeGraph.getNativeHandle(), data));
|
||||||
}
|
}
|
||||||
|
@ -449,6 +453,8 @@ public class PacketCreator {
|
||||||
|
|
||||||
private native long nativeCreateInt32Array(long context, int[] data);
|
private native long nativeCreateInt32Array(long context, int[] data);
|
||||||
|
|
||||||
|
private native long nativeCreateInt32Pair(long context, int first, int second);
|
||||||
|
|
||||||
private native long nativeCreateFloat32Array(long context, float[] data);
|
private native long nativeCreateFloat32Array(long context, float[] data);
|
||||||
|
|
||||||
private native long nativeCreateFloat32Vector(long context, float[] data);
|
private native long nativeCreateFloat32Vector(long context, float[] data);
|
||||||
|
|
|
@ -16,6 +16,7 @@
|
||||||
|
|
||||||
#include <cstring>
|
#include <cstring>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
|
#include <utility>
|
||||||
|
|
||||||
#include "absl/status/status.h"
|
#include "absl/status/status.h"
|
||||||
#include "absl/strings/str_cat.h"
|
#include "absl/strings/str_cat.h"
|
||||||
|
@ -27,6 +28,7 @@
|
||||||
#include "mediapipe/framework/formats/matrix.h"
|
#include "mediapipe/framework/formats/matrix.h"
|
||||||
#include "mediapipe/framework/formats/time_series_header.pb.h"
|
#include "mediapipe/framework/formats/time_series_header.pb.h"
|
||||||
#include "mediapipe/framework/formats/video_stream_header.h"
|
#include "mediapipe/framework/formats/video_stream_header.h"
|
||||||
|
#include "mediapipe/framework/packet.h"
|
||||||
#include "mediapipe/framework/port/core_proto_inc.h"
|
#include "mediapipe/framework/port/core_proto_inc.h"
|
||||||
#include "mediapipe/framework/port/logging.h"
|
#include "mediapipe/framework/port/logging.h"
|
||||||
#include "mediapipe/java/com/google/mediapipe/framework/jni/colorspace.h"
|
#include "mediapipe/java/com/google/mediapipe/framework/jni/colorspace.h"
|
||||||
|
@ -481,6 +483,15 @@ JNIEXPORT jlong JNICALL PACKET_CREATOR_METHOD(nativeCreateInt32Array)(
|
||||||
return CreatePacketWithContext(context, packet);
|
return CreatePacketWithContext(context, packet);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
JNIEXPORT jlong JNICALL PACKET_CREATOR_METHOD(nativeCreateInt32Pair)(
|
||||||
|
JNIEnv* env, jobject thiz, jlong context, jint first, jint second) {
|
||||||
|
static_assert(std::is_same<int32_t, jint>::value, "jint must be int32_t");
|
||||||
|
|
||||||
|
mediapipe::Packet packet = mediapipe::MakePacket<std::pair<int32_t, int32_t>>(
|
||||||
|
std::make_pair(first, second));
|
||||||
|
return CreatePacketWithContext(context, packet);
|
||||||
|
}
|
||||||
|
|
||||||
JNIEXPORT jlong JNICALL PACKET_CREATOR_METHOD(nativeCreateStringFromByteArray)(
|
JNIEXPORT jlong JNICALL PACKET_CREATOR_METHOD(nativeCreateStringFromByteArray)(
|
||||||
JNIEnv* env, jobject thiz, jlong context, jbyteArray data) {
|
JNIEnv* env, jobject thiz, jlong context, jbyteArray data) {
|
||||||
jsize count = env->GetArrayLength(data);
|
jsize count = env->GetArrayLength(data);
|
||||||
|
|
|
@ -118,6 +118,9 @@ JNIEXPORT jlong JNICALL PACKET_CREATOR_METHOD(nativeCreateFloat32Vector)(
|
||||||
JNIEXPORT jlong JNICALL PACKET_CREATOR_METHOD(nativeCreateInt32Array)(
|
JNIEXPORT jlong JNICALL PACKET_CREATOR_METHOD(nativeCreateInt32Array)(
|
||||||
JNIEnv* env, jobject thiz, jlong context, jintArray data);
|
JNIEnv* env, jobject thiz, jlong context, jintArray data);
|
||||||
|
|
||||||
|
JNIEXPORT jlong JNICALL PACKET_CREATOR_METHOD(nativeCreateInt32Pair)(
|
||||||
|
JNIEnv* env, jobject thiz, jlong context, jint first, jint second);
|
||||||
|
|
||||||
JNIEXPORT jlong JNICALL PACKET_CREATOR_METHOD(nativeCreateStringFromByteArray)(
|
JNIEXPORT jlong JNICALL PACKET_CREATOR_METHOD(nativeCreateStringFromByteArray)(
|
||||||
JNIEnv* env, jobject thiz, jlong context, jbyteArray data);
|
JNIEnv* env, jobject thiz, jlong context, jbyteArray data);
|
||||||
|
|
||||||
|
|
|
@ -27,7 +27,7 @@ import java.util.Map;
|
||||||
|
|
||||||
/** The base class of MediaPipe vision tasks. */
|
/** The base class of MediaPipe vision tasks. */
|
||||||
public class BaseVisionTaskApi implements AutoCloseable {
|
public class BaseVisionTaskApi implements AutoCloseable {
|
||||||
private static final long MICROSECONDS_PER_MILLISECOND = 1000;
|
protected static final long MICROSECONDS_PER_MILLISECOND = 1000;
|
||||||
protected final TaskRunner runner;
|
protected final TaskRunner runner;
|
||||||
protected final RunningMode runningMode;
|
protected final RunningMode runningMode;
|
||||||
protected final String imageStreamName;
|
protected final String imageStreamName;
|
||||||
|
@ -69,12 +69,6 @@ public class BaseVisionTaskApi implements AutoCloseable {
|
||||||
*/
|
*/
|
||||||
protected TaskResult processImageData(
|
protected TaskResult processImageData(
|
||||||
MPImage image, ImageProcessingOptions imageProcessingOptions) {
|
MPImage image, ImageProcessingOptions imageProcessingOptions) {
|
||||||
if (runningMode != RunningMode.IMAGE) {
|
|
||||||
throw new MediaPipeException(
|
|
||||||
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
|
||||||
"Task is not initialized with the image mode. Current running mode:"
|
|
||||||
+ runningMode.name());
|
|
||||||
}
|
|
||||||
Map<String, Packet> inputPackets = new HashMap<>();
|
Map<String, Packet> inputPackets = new HashMap<>();
|
||||||
inputPackets.put(imageStreamName, runner.getPacketCreator().createImage(image));
|
inputPackets.put(imageStreamName, runner.getPacketCreator().createImage(image));
|
||||||
if (!normRectStreamName.isEmpty()) {
|
if (!normRectStreamName.isEmpty()) {
|
||||||
|
@ -84,6 +78,23 @@ public class BaseVisionTaskApi implements AutoCloseable {
|
||||||
.getPacketCreator()
|
.getPacketCreator()
|
||||||
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
|
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
|
||||||
}
|
}
|
||||||
|
return processImageData(inputPackets);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A synchronous method to process single image inputs. The call blocks the current thread until a
|
||||||
|
* failure status or a successful result is returned.
|
||||||
|
*
|
||||||
|
* @param inputPackets the maps of input stream names to the input packets.
|
||||||
|
* @throws MediaPipeException if the task is not in the image mode.
|
||||||
|
*/
|
||||||
|
protected TaskResult processImageData(Map<String, Packet> inputPackets) {
|
||||||
|
if (runningMode != RunningMode.IMAGE) {
|
||||||
|
throw new MediaPipeException(
|
||||||
|
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
||||||
|
"Task is not initialized with the image mode. Current running mode:"
|
||||||
|
+ runningMode.name());
|
||||||
|
}
|
||||||
return runner.process(inputPackets);
|
return runner.process(inputPackets);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -99,12 +110,6 @@ public class BaseVisionTaskApi implements AutoCloseable {
|
||||||
*/
|
*/
|
||||||
protected TaskResult processVideoData(
|
protected TaskResult processVideoData(
|
||||||
MPImage image, ImageProcessingOptions imageProcessingOptions, long timestampMs) {
|
MPImage image, ImageProcessingOptions imageProcessingOptions, long timestampMs) {
|
||||||
if (runningMode != RunningMode.VIDEO) {
|
|
||||||
throw new MediaPipeException(
|
|
||||||
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
|
||||||
"Task is not initialized with the video mode. Current running mode:"
|
|
||||||
+ runningMode.name());
|
|
||||||
}
|
|
||||||
Map<String, Packet> inputPackets = new HashMap<>();
|
Map<String, Packet> inputPackets = new HashMap<>();
|
||||||
inputPackets.put(imageStreamName, runner.getPacketCreator().createImage(image));
|
inputPackets.put(imageStreamName, runner.getPacketCreator().createImage(image));
|
||||||
if (!normRectStreamName.isEmpty()) {
|
if (!normRectStreamName.isEmpty()) {
|
||||||
|
@ -114,6 +119,24 @@ public class BaseVisionTaskApi implements AutoCloseable {
|
||||||
.getPacketCreator()
|
.getPacketCreator()
|
||||||
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
|
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
|
||||||
}
|
}
|
||||||
|
return processVideoData(inputPackets, timestampMs * MICROSECONDS_PER_MILLISECOND);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A synchronous method to process continuous video frames. The call blocks the current thread
|
||||||
|
* until a failure status or a successful result is returned.
|
||||||
|
*
|
||||||
|
* @param inputPackets the maps of input stream names to the input packets.
|
||||||
|
* @param timestampMs the corresponding timestamp of the input image in milliseconds.
|
||||||
|
* @throws MediaPipeException if the task is not in the video mode.
|
||||||
|
*/
|
||||||
|
protected TaskResult processVideoData(Map<String, Packet> inputPackets, long timestampMs) {
|
||||||
|
if (runningMode != RunningMode.VIDEO) {
|
||||||
|
throw new MediaPipeException(
|
||||||
|
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
||||||
|
"Task is not initialized with the video mode. Current running mode:"
|
||||||
|
+ runningMode.name());
|
||||||
|
}
|
||||||
return runner.process(inputPackets, timestampMs * MICROSECONDS_PER_MILLISECOND);
|
return runner.process(inputPackets, timestampMs * MICROSECONDS_PER_MILLISECOND);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -129,12 +152,6 @@ public class BaseVisionTaskApi implements AutoCloseable {
|
||||||
*/
|
*/
|
||||||
protected void sendLiveStreamData(
|
protected void sendLiveStreamData(
|
||||||
MPImage image, ImageProcessingOptions imageProcessingOptions, long timestampMs) {
|
MPImage image, ImageProcessingOptions imageProcessingOptions, long timestampMs) {
|
||||||
if (runningMode != RunningMode.LIVE_STREAM) {
|
|
||||||
throw new MediaPipeException(
|
|
||||||
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
|
||||||
"Task is not initialized with the live stream mode. Current running mode:"
|
|
||||||
+ runningMode.name());
|
|
||||||
}
|
|
||||||
Map<String, Packet> inputPackets = new HashMap<>();
|
Map<String, Packet> inputPackets = new HashMap<>();
|
||||||
inputPackets.put(imageStreamName, runner.getPacketCreator().createImage(image));
|
inputPackets.put(imageStreamName, runner.getPacketCreator().createImage(image));
|
||||||
if (!normRectStreamName.isEmpty()) {
|
if (!normRectStreamName.isEmpty()) {
|
||||||
|
@ -144,6 +161,24 @@ public class BaseVisionTaskApi implements AutoCloseable {
|
||||||
.getPacketCreator()
|
.getPacketCreator()
|
||||||
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
|
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
|
||||||
}
|
}
|
||||||
|
sendLiveStreamData(inputPackets, timestampMs * MICROSECONDS_PER_MILLISECOND);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* An asynchronous method to send live stream data to the {@link TaskRunner}. The results will be
|
||||||
|
* available in the user-defined result listener.
|
||||||
|
*
|
||||||
|
* @param inputPackets the maps of input stream names to the input packets.
|
||||||
|
* @param timestampMs the corresponding timestamp of the input image in milliseconds.
|
||||||
|
* @throws MediaPipeException if the task is not in the stream mode.
|
||||||
|
*/
|
||||||
|
protected void sendLiveStreamData(Map<String, Packet> inputPackets, long timestampMs) {
|
||||||
|
if (runningMode != RunningMode.LIVE_STREAM) {
|
||||||
|
throw new MediaPipeException(
|
||||||
|
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
||||||
|
"Task is not initialized with the live stream mode. Current running mode:"
|
||||||
|
+ runningMode.name());
|
||||||
|
}
|
||||||
runner.send(inputPackets, timestampMs * MICROSECONDS_PER_MILLISECOND);
|
runner.send(inputPackets, timestampMs * MICROSECONDS_PER_MILLISECOND);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -43,7 +43,9 @@ import java.nio.ByteBuffer;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
import java.util.Collections;
|
import java.util.Collections;
|
||||||
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
import java.util.Optional;
|
import java.util.Optional;
|
||||||
import java.util.function.BiFunction;
|
import java.util.function.BiFunction;
|
||||||
|
|
||||||
|
@ -77,9 +79,13 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
private static final String TAG = ImageSegmenter.class.getSimpleName();
|
private static final String TAG = ImageSegmenter.class.getSimpleName();
|
||||||
private static final String IMAGE_IN_STREAM_NAME = "image_in";
|
private static final String IMAGE_IN_STREAM_NAME = "image_in";
|
||||||
private static final String NORM_RECT_IN_STREAM_NAME = "norm_rect_in";
|
private static final String NORM_RECT_IN_STREAM_NAME = "norm_rect_in";
|
||||||
|
private static final String OUTPUT_SIZE_IN_STREAM_NAME = "output_size_in";
|
||||||
private static final List<String> INPUT_STREAMS =
|
private static final List<String> INPUT_STREAMS =
|
||||||
Collections.unmodifiableList(
|
Collections.unmodifiableList(
|
||||||
Arrays.asList("IMAGE:" + IMAGE_IN_STREAM_NAME, "NORM_RECT:" + NORM_RECT_IN_STREAM_NAME));
|
Arrays.asList(
|
||||||
|
"IMAGE:" + IMAGE_IN_STREAM_NAME,
|
||||||
|
"NORM_RECT:" + NORM_RECT_IN_STREAM_NAME,
|
||||||
|
"OUTPUT_SIZE:" + OUTPUT_SIZE_IN_STREAM_NAME));
|
||||||
private static final String TASK_GRAPH_NAME =
|
private static final String TASK_GRAPH_NAME =
|
||||||
"mediapipe.tasks.vision.image_segmenter.ImageSegmenterGraph";
|
"mediapipe.tasks.vision.image_segmenter.ImageSegmenterGraph";
|
||||||
private static final String TENSORS_TO_SEGMENTATION_CALCULATOR_NAME =
|
private static final String TENSORS_TO_SEGMENTATION_CALCULATOR_NAME =
|
||||||
|
@ -238,6 +244,7 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
this.hasResultListener = hasResultListener;
|
this.hasResultListener = hasResultListener;
|
||||||
populateLabels();
|
populateLabels();
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Populate the labelmap in TensorsToSegmentationCalculator to labels field.
|
* Populate the labelmap in TensorsToSegmentationCalculator to labels field.
|
||||||
*
|
*
|
||||||
|
@ -275,9 +282,9 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Performs image segmentation on the provided single image with default image processing options,
|
* Performs image segmentation on the provided single image with default image processing options,
|
||||||
* i.e. without any rotation applied. Only use this method when the {@link ImageSegmenter} is
|
* i.e. without any rotation applied. The output mask has the same size as the input image. Only
|
||||||
* created with {@link RunningMode.IMAGE}. TODO update java doc for input image
|
* use this method when the {@link ImageSegmenter} is created with {@link RunningMode.IMAGE}.
|
||||||
* format.
|
* TODO update java doc for input image format.
|
||||||
*
|
*
|
||||||
* <p>{@link ImageSegmenter} supports the following color space types:
|
* <p>{@link ImageSegmenter} supports the following color space types:
|
||||||
*
|
*
|
||||||
|
@ -294,9 +301,9 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Performs image segmentation on the provided single image. Only use this method when the {@link
|
* Performs image segmentation on the provided single image. The output mask has the same size as
|
||||||
* ImageSegmenter} is created with {@link RunningMode.IMAGE}. TODO update java doc
|
* the input image. Only use this method when the {@link ImageSegmenter} is created with {@link
|
||||||
* for input image format.
|
* RunningMode.IMAGE}. TODO update java doc for input image format.
|
||||||
*
|
*
|
||||||
* <p>{@link ImageSegmenter} supports the following color space types:
|
* <p>{@link ImageSegmenter} supports the following color space types:
|
||||||
*
|
*
|
||||||
|
@ -316,21 +323,47 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
*/
|
*/
|
||||||
public ImageSegmenterResult segment(
|
public ImageSegmenterResult segment(
|
||||||
MPImage image, ImageProcessingOptions imageProcessingOptions) {
|
MPImage image, ImageProcessingOptions imageProcessingOptions) {
|
||||||
|
return segment(
|
||||||
|
image,
|
||||||
|
SegmentationOptions.builder()
|
||||||
|
.setOutputWidth(image.getWidth())
|
||||||
|
.setOutputHeight(image.getHeight())
|
||||||
|
.setImageProcessingOptions(imageProcessingOptions)
|
||||||
|
.build());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Performs image segmentation on the provided single image. Only use this method when the {@link
|
||||||
|
* ImageSegmenter} is created with {@link RunningMode.IMAGE}. TODO update java doc
|
||||||
|
* for input image format.
|
||||||
|
*
|
||||||
|
* <p>{@link ImageSegmenter} supports the following color space types:
|
||||||
|
*
|
||||||
|
* <ul>
|
||||||
|
* <li>{@link Bitmap.Config.ARGB_8888}
|
||||||
|
* </ul>
|
||||||
|
*
|
||||||
|
* @param image a MediaPipe {@link MPImage} object for processing.
|
||||||
|
* @param segmentationOptions the {@link SegmentationOptions} used to configure the runtime
|
||||||
|
* behavior of the {@link ImageSegmenter}.
|
||||||
|
* @throws MediaPipeException if there is an internal error. Or if {@link ImageSegmenter} is
|
||||||
|
* created with a {@link ResultListener}.
|
||||||
|
*/
|
||||||
|
public ImageSegmenterResult segment(MPImage image, SegmentationOptions segmentationOptions) {
|
||||||
if (hasResultListener) {
|
if (hasResultListener) {
|
||||||
throw new MediaPipeException(
|
throw new MediaPipeException(
|
||||||
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
||||||
"ResultListener is provided in the ImageSegmenterOptions, but this method will return an"
|
"ResultListener is provided in the ImageSegmenterOptions, but this method will return an"
|
||||||
+ " ImageSegmentationResult.");
|
+ " ImageSegmentationResult.");
|
||||||
}
|
}
|
||||||
validateImageProcessingOptions(imageProcessingOptions);
|
return (ImageSegmenterResult) processImageData(buildInputPackets(image, segmentationOptions));
|
||||||
return (ImageSegmenterResult) processImageData(image, imageProcessingOptions);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Performs image segmentation on the provided single image with default image processing options,
|
* Performs image segmentation on the provided single image with default image processing options,
|
||||||
* i.e. without any rotation applied, and provides zero-copied results via {@link ResultListener}
|
* i.e. without any rotation applied, and provides zero-copied results via {@link ResultListener}
|
||||||
* in {@link ImageSegmenterOptions}. Only use this method when the {@link ImageSegmenter} is
|
* in {@link ImageSegmenterOptions}. The output mask has the same size as the input image. Only
|
||||||
* created with {@link RunningMode.IMAGE}.
|
* use this method when the {@link ImageSegmenter} is created with {@link RunningMode.IMAGE}.
|
||||||
*
|
*
|
||||||
* <p>TODO update java doc for input image format.
|
* <p>TODO update java doc for input image format.
|
||||||
*
|
*
|
||||||
|
@ -341,8 +374,6 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
* </ul>
|
* </ul>
|
||||||
*
|
*
|
||||||
* @param image a MediaPipe {@link MPImage} object for processing.
|
* @param image a MediaPipe {@link MPImage} object for processing.
|
||||||
* @throws IllegalArgumentException if the {@link ImageProcessingOptions} specify a
|
|
||||||
* region-of-interest.
|
|
||||||
* @throws MediaPipeException if there is an internal error. Or if {@link ImageSegmenter} is not
|
* @throws MediaPipeException if there is an internal error. Or if {@link ImageSegmenter} is not
|
||||||
* created with {@link ResultListener} set in {@link ImageSegmenterOptions}.
|
* created with {@link ResultListener} set in {@link ImageSegmenterOptions}.
|
||||||
*/
|
*/
|
||||||
|
@ -352,8 +383,9 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Performs image segmentation on the provided single image, and provides zero-copied results via
|
* Performs image segmentation on the provided single image, and provides zero-copied results via
|
||||||
* {@link ResultListener} in {@link ImageSegmenterOptions}. Only use this method when the {@link
|
* {@link ResultListener} in {@link ImageSegmenterOptions}. The output mask has the same size as
|
||||||
* ImageSegmenter} is created with {@link RunningMode.IMAGE}.
|
* the input image. Only use this method when the {@link ImageSegmenter} is created with {@link
|
||||||
|
* RunningMode.IMAGE}.
|
||||||
*
|
*
|
||||||
* <p>TODO update java doc for input image format.
|
* <p>TODO update java doc for input image format.
|
||||||
*
|
*
|
||||||
|
@ -375,21 +407,53 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
*/
|
*/
|
||||||
public void segmentWithResultListener(
|
public void segmentWithResultListener(
|
||||||
MPImage image, ImageProcessingOptions imageProcessingOptions) {
|
MPImage image, ImageProcessingOptions imageProcessingOptions) {
|
||||||
|
segmentWithResultListener(
|
||||||
|
image,
|
||||||
|
SegmentationOptions.builder()
|
||||||
|
.setOutputWidth(image.getWidth())
|
||||||
|
.setOutputHeight(image.getHeight())
|
||||||
|
.setImageProcessingOptions(imageProcessingOptions)
|
||||||
|
.build());
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Performs image segmentation on the provided single image, and provides zero-copied results via
|
||||||
|
* {@link ResultListener} in {@link ImageSegmenterOptions}. Only use this method when the {@link
|
||||||
|
* ImageSegmenter} is created with {@link RunningMode.IMAGE}.
|
||||||
|
*
|
||||||
|
* <p>TODO update java doc for input image format.
|
||||||
|
*
|
||||||
|
* <p>{@link ImageSegmenter} supports the following color space types:
|
||||||
|
*
|
||||||
|
* <ul>
|
||||||
|
* <li>{@link Bitmap.Config.ARGB_8888}
|
||||||
|
* </ul>
|
||||||
|
*
|
||||||
|
* @param image a MediaPipe {@link MPImage} object for processing.
|
||||||
|
* @param segmentationOptions the {@link SegmentationOptions} used to configure the runtime
|
||||||
|
* behavior of the {@link ImageSegmenter}.
|
||||||
|
* @param imageProcessingOptions the {@link ImageProcessingOptions} specifying how to process the
|
||||||
|
* input image before running inference. Note that region-of-interest is <b>not</b> supported
|
||||||
|
* by this task: specifying {@link ImageProcessingOptions#regionOfInterest()} will result in
|
||||||
|
* this method throwing an IllegalArgumentException.
|
||||||
|
* @throws MediaPipeException if there is an internal error. Or if {@link ImageSegmenter} is not
|
||||||
|
* created with {@link ResultListener} set in {@link ImageSegmenterOptions}.
|
||||||
|
*/
|
||||||
|
public void segmentWithResultListener(MPImage image, SegmentationOptions segmentationOptions) {
|
||||||
if (!hasResultListener) {
|
if (!hasResultListener) {
|
||||||
throw new MediaPipeException(
|
throw new MediaPipeException(
|
||||||
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
||||||
"ResultListener is not set in the ImageSegmenterOptions, but this method expects a"
|
"ResultListener is not set in the ImageSegmenterOptions, but this method expects a"
|
||||||
+ " ResultListener to process ImageSegmentationResult.");
|
+ " ResultListener to process ImageSegmentationResult.");
|
||||||
}
|
}
|
||||||
validateImageProcessingOptions(imageProcessingOptions);
|
|
||||||
ImageSegmenterResult unused =
|
ImageSegmenterResult unused =
|
||||||
(ImageSegmenterResult) processImageData(image, imageProcessingOptions);
|
(ImageSegmenterResult) processImageData(buildInputPackets(image, segmentationOptions));
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Performs image segmentation on the provided video frame with default image processing options,
|
* Performs image segmentation on the provided video frame with default image processing options,
|
||||||
* i.e. without any rotation applied. Only use this method when the {@link ImageSegmenter} is
|
* i.e. without any rotation applied. The output mask has the same size as the input image. Only
|
||||||
* created with {@link RunningMode.VIDEO}.
|
* use this method when the {@link ImageSegmenter} is created with {@link RunningMode.VIDEO}.
|
||||||
*
|
*
|
||||||
* <p>It's required to provide the video frame's timestamp (in milliseconds). The input timestamps
|
* <p>It's required to provide the video frame's timestamp (in milliseconds). The input timestamps
|
||||||
* must be monotonically increasing.
|
* must be monotonically increasing.
|
||||||
|
@ -410,8 +474,9 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Performs image segmentation on the provided video frame. Only use this method when the {@link
|
* Performs image segmentation on the provided video frame. The output mask has the same size as
|
||||||
* ImageSegmenter} is created with {@link RunningMode.VIDEO}.
|
* the input image. Only use this method when the {@link ImageSegmenter} is created with {@link
|
||||||
|
* RunningMode.VIDEO}.
|
||||||
*
|
*
|
||||||
* <p>It's required to provide the video frame's timestamp (in milliseconds). The input timestamps
|
* <p>It's required to provide the video frame's timestamp (in milliseconds). The input timestamps
|
||||||
* must be monotonically increasing.
|
* must be monotonically increasing.
|
||||||
|
@ -435,21 +500,53 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
*/
|
*/
|
||||||
public ImageSegmenterResult segmentForVideo(
|
public ImageSegmenterResult segmentForVideo(
|
||||||
MPImage image, ImageProcessingOptions imageProcessingOptions, long timestampMs) {
|
MPImage image, ImageProcessingOptions imageProcessingOptions, long timestampMs) {
|
||||||
|
return segmentForVideo(
|
||||||
|
image,
|
||||||
|
SegmentationOptions.builder()
|
||||||
|
.setOutputWidth(image.getWidth())
|
||||||
|
.setOutputHeight(image.getHeight())
|
||||||
|
.setImageProcessingOptions(imageProcessingOptions)
|
||||||
|
.build(),
|
||||||
|
timestampMs);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Performs image segmentation on the provided video frame. Only use this method when the {@link
|
||||||
|
* ImageSegmenter} is created with {@link RunningMode.VIDEO}.
|
||||||
|
*
|
||||||
|
* <p>It's required to provide the video frame's timestamp (in milliseconds). The input timestamps
|
||||||
|
* must be monotonically increasing.
|
||||||
|
*
|
||||||
|
* <p>{@link ImageSegmenter} supports the following color space types:
|
||||||
|
*
|
||||||
|
* <ul>
|
||||||
|
* <li>{@link Bitmap.Config.ARGB_8888}
|
||||||
|
* </ul>
|
||||||
|
*
|
||||||
|
* @param image a MediaPipe {@link MPImage} object for processing.
|
||||||
|
* @param segmentationOptions the {@link SegmentationOptions} used to configure the runtime
|
||||||
|
* behavior of the {@link ImageSegmenter}.
|
||||||
|
* @param timestampMs the input timestamp (in milliseconds).
|
||||||
|
* @throws MediaPipeException if there is an internal error. Or if {@link ImageSegmenter} is
|
||||||
|
* created with a {@link ResultListener}.
|
||||||
|
*/
|
||||||
|
public ImageSegmenterResult segmentForVideo(
|
||||||
|
MPImage image, SegmentationOptions segmentationOptions, long timestampMs) {
|
||||||
if (hasResultListener) {
|
if (hasResultListener) {
|
||||||
throw new MediaPipeException(
|
throw new MediaPipeException(
|
||||||
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
||||||
"ResultListener is provided in the ImageSegmenterOptions, but this method will return an"
|
"ResultListener is provided in the ImageSegmenterOptions, but this method will return an"
|
||||||
+ " ImageSegmentationResult.");
|
+ " ImageSegmentationResult.");
|
||||||
}
|
}
|
||||||
validateImageProcessingOptions(imageProcessingOptions);
|
return (ImageSegmenterResult)
|
||||||
return (ImageSegmenterResult) processVideoData(image, imageProcessingOptions, timestampMs);
|
processVideoData(buildInputPackets(image, segmentationOptions), timestampMs);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Performs image segmentation on the provided video frame with default image processing options,
|
* Performs image segmentation on the provided video frame with default image processing options,
|
||||||
* i.e. without any rotation applied, and provides zero-copied results via {@link ResultListener}
|
* i.e. without any rotation applied, and provides zero-copied results via {@link ResultListener}
|
||||||
* in {@link ImageSegmenterOptions}. Only use this method when the {@link ImageSegmenter} is
|
* in {@link ImageSegmenterOptions}. The output mask has the same size as the input image. Only
|
||||||
* created with {@link RunningMode.VIDEO}.
|
* use this method when the {@link ImageSegmenter} is created with {@link RunningMode.VIDEO}.
|
||||||
*
|
*
|
||||||
* <p>It's required to provide the video frame's timestamp (in milliseconds). The input timestamps
|
* <p>It's required to provide the video frame's timestamp (in milliseconds). The input timestamps
|
||||||
* must be monotonically increasing.
|
* must be monotonically increasing.
|
||||||
|
@ -469,6 +566,40 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
segmentForVideoWithResultListener(image, ImageProcessingOptions.builder().build(), timestampMs);
|
segmentForVideoWithResultListener(image, ImageProcessingOptions.builder().build(), timestampMs);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Performs image segmentation on the provided video frame, and provides zero-copied results via
|
||||||
|
* {@link ResultListener} in {@link ImageSegmenterOptions}. The output mask has the same size as
|
||||||
|
* the input image. Only use this method when the {@link ImageSegmenter} is created with {@link
|
||||||
|
* RunningMode.VIDEO}.
|
||||||
|
*
|
||||||
|
* <p>It's required to provide the video frame's timestamp (in milliseconds). The input timestamps
|
||||||
|
* must be monotonically increasing.
|
||||||
|
*
|
||||||
|
* <p>{@link ImageSegmenter} supports the following color space types:
|
||||||
|
*
|
||||||
|
* <ul>
|
||||||
|
* <li>{@link Bitmap.Config.ARGB_8888}
|
||||||
|
* </ul>
|
||||||
|
*
|
||||||
|
* @param image a MediaPipe {@link MPImage} object for processing.
|
||||||
|
* @param timestampMs the input timestamp (in milliseconds).
|
||||||
|
* @throws IllegalArgumentException if the {@link ImageProcessingOptions} specify a
|
||||||
|
* region-of-interest.
|
||||||
|
* @throws MediaPipeException if there is an internal error. Or if {@link ImageSegmenter} is not
|
||||||
|
* created with {@link ResultListener} set in {@link ImageSegmenterOptions}.
|
||||||
|
*/
|
||||||
|
public void segmentForVideoWithResultListener(
|
||||||
|
MPImage image, ImageProcessingOptions imageProcessingOptions, long timestampMs) {
|
||||||
|
segmentForVideoWithResultListener(
|
||||||
|
image,
|
||||||
|
SegmentationOptions.builder()
|
||||||
|
.setOutputWidth(image.getWidth())
|
||||||
|
.setOutputHeight(image.getHeight())
|
||||||
|
.setImageProcessingOptions(imageProcessingOptions)
|
||||||
|
.build(),
|
||||||
|
timestampMs);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Performs image segmentation on the provided video frame, and provides zero-copied results via
|
* Performs image segmentation on the provided video frame, and provides zero-copied results via
|
||||||
* {@link ResultListener} in {@link ImageSegmenterOptions}. Only use this method when the {@link
|
* {@link ResultListener} in {@link ImageSegmenterOptions}. Only use this method when the {@link
|
||||||
|
@ -484,28 +615,31 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
* </ul>
|
* </ul>
|
||||||
*
|
*
|
||||||
* @param image a MediaPipe {@link MPImage} object for processing.
|
* @param image a MediaPipe {@link MPImage} object for processing.
|
||||||
|
* @param segmentationOptions the {@link SegmentationOptions} used to configure the runtime
|
||||||
|
* behavior of the {@link ImageSegmenter}.
|
||||||
* @param timestampMs the input timestamp (in milliseconds).
|
* @param timestampMs the input timestamp (in milliseconds).
|
||||||
* @throws MediaPipeException if there is an internal error. Or if {@link ImageSegmenter} is not
|
* @throws MediaPipeException if there is an internal error. Or if {@link ImageSegmenter} is not
|
||||||
* created with {@link ResultListener} set in {@link ImageSegmenterOptions}.
|
* created with {@link ResultListener} set in {@link ImageSegmenterOptions}.
|
||||||
*/
|
*/
|
||||||
public void segmentForVideoWithResultListener(
|
public void segmentForVideoWithResultListener(
|
||||||
MPImage image, ImageProcessingOptions imageProcessingOptions, long timestampMs) {
|
MPImage image, SegmentationOptions segmentationOptions, long timestampMs) {
|
||||||
if (!hasResultListener) {
|
if (!hasResultListener) {
|
||||||
throw new MediaPipeException(
|
throw new MediaPipeException(
|
||||||
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
MediaPipeException.StatusCode.FAILED_PRECONDITION.ordinal(),
|
||||||
"ResultListener is not set in the ImageSegmenterOptions, but this method expects a"
|
"ResultListener is not set in the ImageSegmenterOptions, but this method expects a"
|
||||||
+ " ResultListener to process ImageSegmentationResult.");
|
+ " ResultListener to process ImageSegmentationResult.");
|
||||||
}
|
}
|
||||||
validateImageProcessingOptions(imageProcessingOptions);
|
|
||||||
ImageSegmenterResult unused =
|
ImageSegmenterResult unused =
|
||||||
(ImageSegmenterResult) processVideoData(image, imageProcessingOptions, timestampMs);
|
(ImageSegmenterResult)
|
||||||
|
processVideoData(buildInputPackets(image, segmentationOptions), timestampMs);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Sends live image data to perform image segmentation with default image processing options, i.e.
|
* Sends live image data to perform image segmentation with default image processing options, i.e.
|
||||||
* without any rotation applied, and the results will be available via the {@link ResultListener}
|
* without any rotation applied, and the results will be available via the {@link ResultListener}
|
||||||
* provided in the {@link ImageSegmenterOptions}. Only use this method when the {@link
|
* provided in the {@link ImageSegmenterOptions}. The output mask has the same size as the input
|
||||||
* ImageSegmenter } is created with {@link RunningMode.LIVE_STREAM}.
|
* image. Only use this method when the {@link ImageSegmenter } is created with {@link
|
||||||
|
* RunningMode.LIVE_STREAM}.
|
||||||
*
|
*
|
||||||
* <p>It's required to provide a timestamp (in milliseconds) to indicate when the input image is
|
* <p>It's required to provide a timestamp (in milliseconds) to indicate when the input image is
|
||||||
* sent to the image segmenter. The input timestamps must be monotonically increasing.
|
* sent to the image segmenter. The input timestamps must be monotonically increasing.
|
||||||
|
@ -526,8 +660,9 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Sends live image data to perform image segmentation, and the results will be available via the
|
* Sends live image data to perform image segmentation, and the results will be available via the
|
||||||
* {@link ResultListener} provided in the {@link ImageSegmenterOptions}. Only use this method when
|
* {@link ResultListener} provided in the {@link ImageSegmenterOptions}. The output mask has the
|
||||||
* the {@link ImageSegmenter} is created with {@link RunningMode.LIVE_STREAM}.
|
* same size as the input image. Only use this method when the {@link ImageSegmenter} is created
|
||||||
|
* with {@link RunningMode.LIVE_STREAM}.
|
||||||
*
|
*
|
||||||
* <p>It's required to provide a timestamp (in milliseconds) to indicate when the input image is
|
* <p>It's required to provide a timestamp (in milliseconds) to indicate when the input image is
|
||||||
* sent to the image segmenter. The input timestamps must be monotonically increasing.
|
* sent to the image segmenter. The input timestamps must be monotonically increasing.
|
||||||
|
@ -550,8 +685,39 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
*/
|
*/
|
||||||
public void segmentAsync(
|
public void segmentAsync(
|
||||||
MPImage image, ImageProcessingOptions imageProcessingOptions, long timestampMs) {
|
MPImage image, ImageProcessingOptions imageProcessingOptions, long timestampMs) {
|
||||||
validateImageProcessingOptions(imageProcessingOptions);
|
segmentAsync(
|
||||||
sendLiveStreamData(image, imageProcessingOptions, timestampMs);
|
image,
|
||||||
|
SegmentationOptions.builder()
|
||||||
|
.setOutputWidth(image.getWidth())
|
||||||
|
.setOutputHeight(image.getHeight())
|
||||||
|
.setImageProcessingOptions(imageProcessingOptions)
|
||||||
|
.build(),
|
||||||
|
timestampMs);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Sends live image data to perform image segmentation, and the results will be available via the
|
||||||
|
* {@link ResultListener} provided in the {@link ImageSegmenterOptions}. Only use this method when
|
||||||
|
* the {@link ImageSegmenter} is created with {@link RunningMode.LIVE_STREAM}.
|
||||||
|
*
|
||||||
|
* <p>It's required to provide a timestamp (in milliseconds) to indicate when the input image is
|
||||||
|
* sent to the image segmenter. The input timestamps must be monotonically increasing.
|
||||||
|
*
|
||||||
|
* <p>{@link ImageSegmenter} supports the following color space types:
|
||||||
|
*
|
||||||
|
* <ul>
|
||||||
|
* <li>{@link Bitmap.Config.ARGB_8888}
|
||||||
|
* </ul>
|
||||||
|
*
|
||||||
|
* @param image a MediaPipe {@link MPImage} object for processing.
|
||||||
|
* @param segmentationOptions the {@link SegmentationOptions} used to configure the runtime
|
||||||
|
* behavior of the {@link ImageSegmenter}.
|
||||||
|
* @param timestampMs the input timestamp (in milliseconds).
|
||||||
|
* @throws MediaPipeException if there is an internal error.
|
||||||
|
*/
|
||||||
|
public void segmentAsync(
|
||||||
|
MPImage image, SegmentationOptions segmentationOptions, long timestampMs) {
|
||||||
|
sendLiveStreamData(buildInputPackets(image, segmentationOptions), timestampMs);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -565,6 +731,56 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
return labels;
|
return labels;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/** Options for configuring runtime behavior of {@link ImageSegmenter}. */
|
||||||
|
@AutoValue
|
||||||
|
public abstract static class SegmentationOptions {
|
||||||
|
|
||||||
|
/** Builder fo {@link SegmentationOptions} */
|
||||||
|
@AutoValue.Builder
|
||||||
|
public abstract static class Builder {
|
||||||
|
|
||||||
|
/** Set the width of the output segmentation masks. */
|
||||||
|
public abstract Builder setOutputWidth(int value);
|
||||||
|
|
||||||
|
/** Set the height of the output segmentation masks. */
|
||||||
|
public abstract Builder setOutputHeight(int value);
|
||||||
|
|
||||||
|
/** Set the image processing options. */
|
||||||
|
public abstract Builder setImageProcessingOptions(ImageProcessingOptions value);
|
||||||
|
|
||||||
|
abstract SegmentationOptions autoBuild();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Validates and builds the {@link SegmentationOptions} instance.
|
||||||
|
*
|
||||||
|
* @throws IllegalArgumentException if the {@link ImageProcessingOptions} specify a
|
||||||
|
* region-of-interest.
|
||||||
|
*/
|
||||||
|
public final SegmentationOptions build() {
|
||||||
|
SegmentationOptions options = autoBuild();
|
||||||
|
if (options.outputWidth() <= 0 || options.outputHeight() <= 0) {
|
||||||
|
throw new IllegalArgumentException(
|
||||||
|
"Both outputWidth and outputHeight must be larger than 0.");
|
||||||
|
}
|
||||||
|
if (options.imageProcessingOptions().regionOfInterest().isPresent()) {
|
||||||
|
throw new IllegalArgumentException("ImageSegmenter doesn't support region-of-interest.");
|
||||||
|
}
|
||||||
|
return options;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
abstract int outputWidth();
|
||||||
|
|
||||||
|
abstract int outputHeight();
|
||||||
|
|
||||||
|
abstract ImageProcessingOptions imageProcessingOptions();
|
||||||
|
|
||||||
|
static Builder builder() {
|
||||||
|
return new AutoValue_ImageSegmenter_SegmentationOptions.Builder()
|
||||||
|
.setImageProcessingOptions(ImageProcessingOptions.builder().build());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/** Options for setting up an {@link ImageSegmenter}. */
|
/** Options for setting up an {@link ImageSegmenter}. */
|
||||||
@AutoValue
|
@AutoValue
|
||||||
public abstract static class ImageSegmenterOptions extends TaskOptions {
|
public abstract static class ImageSegmenterOptions extends TaskOptions {
|
||||||
|
@ -680,14 +896,24 @@ public final class ImageSegmenter extends BaseVisionTaskApi {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
private Map<String, Packet> buildInputPackets(
|
||||||
* Validates that the provided {@link ImageProcessingOptions} doesn't contain a
|
MPImage image, SegmentationOptions segmentationOptions) {
|
||||||
* region-of-interest.
|
Map<String, Packet> inputPackets = new HashMap<>();
|
||||||
*/
|
inputPackets.put(imageStreamName, runner.getPacketCreator().createImage(image));
|
||||||
private static void validateImageProcessingOptions(
|
inputPackets.put(
|
||||||
ImageProcessingOptions imageProcessingOptions) {
|
OUTPUT_SIZE_IN_STREAM_NAME,
|
||||||
if (imageProcessingOptions.regionOfInterest().isPresent()) {
|
runner
|
||||||
throw new IllegalArgumentException("ImageSegmenter doesn't support region-of-interest.");
|
.getPacketCreator()
|
||||||
}
|
.createInt32Pair(
|
||||||
|
segmentationOptions.outputWidth(), segmentationOptions.outputHeight()));
|
||||||
|
if (!normRectStreamName.isEmpty()) {
|
||||||
|
inputPackets.put(
|
||||||
|
normRectStreamName,
|
||||||
|
runner
|
||||||
|
.getPacketCreator()
|
||||||
|
.createProto(
|
||||||
|
convertToNormalizedRect(segmentationOptions.imageProcessingOptions(), image)));
|
||||||
|
}
|
||||||
|
return inputPackets;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -31,6 +31,7 @@ import com.google.mediapipe.framework.image.MPImage;
|
||||||
import com.google.mediapipe.tasks.core.BaseOptions;
|
import com.google.mediapipe.tasks.core.BaseOptions;
|
||||||
import com.google.mediapipe.tasks.vision.core.RunningMode;
|
import com.google.mediapipe.tasks.vision.core.RunningMode;
|
||||||
import com.google.mediapipe.tasks.vision.imagesegmenter.ImageSegmenter.ImageSegmenterOptions;
|
import com.google.mediapipe.tasks.vision.imagesegmenter.ImageSegmenter.ImageSegmenterOptions;
|
||||||
|
import com.google.mediapipe.tasks.vision.imagesegmenter.ImageSegmenter.SegmentationOptions;
|
||||||
import java.io.InputStream;
|
import java.io.InputStream;
|
||||||
import java.nio.ByteBuffer;
|
import java.nio.ByteBuffer;
|
||||||
import java.nio.FloatBuffer;
|
import java.nio.FloatBuffer;
|
||||||
|
|
Loading…
Reference in New Issue
Block a user