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

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