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

internal change.

Browse Source 
PiperOrigin-RevId: 529181374
This commit is contained in:
Yuqi Li 2023-05-03 13:32:20 -07:00 committed by Copybara-Service
parent 606b83ac65
commit e428bdb7e8
1 changed files with 21 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
mediapipe/tasks/java/com/google/mediapipe/tasks/vision/core/BaseVisionTaskApi.java
View File

@ -77,11 +77,13 @@ public class BaseVisionTaskApi implements AutoCloseable {
}
Map<String, Packet> inputPackets = new HashMap<>();
inputPackets.put(imageStreamName, runner.getPacketCreator().createImage(image));
inputPackets.put(
normRectStreamName,
runner
.getPacketCreator()
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
if (!normRectStreamName.isEmpty()) {
inputPackets.put(
normRectStreamName,
runner
.getPacketCreator()
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
}
return runner.process(inputPackets);
}
@ -105,11 +107,13 @@ public class BaseVisionTaskApi implements AutoCloseable {
}
Map<String, Packet> inputPackets = new HashMap<>();
inputPackets.put(imageStreamName, runner.getPacketCreator().createImage(image));
inputPackets.put(
normRectStreamName,
runner
.getPacketCreator()
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
if (!normRectStreamName.isEmpty()) {
inputPackets.put(
normRectStreamName,
runner
.getPacketCreator()
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
}
return runner.process(inputPackets, timestampMs * MICROSECONDS_PER_MILLISECOND);
}
@ -133,11 +137,13 @@ public class BaseVisionTaskApi implements AutoCloseable {
}
Map<String, Packet> inputPackets = new HashMap<>();
inputPackets.put(imageStreamName, runner.getPacketCreator().createImage(image));
inputPackets.put(
normRectStreamName,
runner
.getPacketCreator()
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
if (!normRectStreamName.isEmpty()) {
inputPackets.put(
normRectStreamName,
runner
.getPacketCreator()
.createProto(convertToNormalizedRect(imageProcessingOptions, image)));
}
runner.send(inputPackets, timestampMs * MICROSECONDS_PER_MILLISECOND);
}