talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Project import generated by Copybara.

Browse Source 
GitOrigin-RevId: 6a704ded0bf489614797082e7e7cda1068477ef5
This commit is contained in:
MediaPipe Team
2021-03-31 20:33:42 -04:00
committed by chuoling
parent 7c331ad58b
commit ecb5b5f44a
84 changed files with 1015 additions and 968 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/framework/api2/builder_test.cc
+10 -10
View File
@@ -27,7 +27,7 @@ TEST(BuilderTest, BuildGraph) {
bar.Out("OUT").SetName("out") >> graph.Out("OUT");
CalculatorGraphConfig expected =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "IN:base"
input_side_packet: "SIDE:side"
output_stream: "OUT:out"
@@ -42,7 +42,7 @@ TEST(BuilderTest, BuildGraph) {
input_stream: "IN:__stream_0"
output_stream: "OUT:out"
}
)");
)pb");
EXPECT_THAT(graph.GetConfig(), EqualsProto(expected));
}
@@ -92,7 +92,7 @@ TEST(BuilderTest, FanOut) {
adder.Out("OUT").SetName("out") >> graph.Out("OUT");
CalculatorGraphConfig expected =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "IN:base"
output_stream: "OUT:out"
node {
@@ -106,7 +106,7 @@ TEST(BuilderTest, FanOut) {
input_stream: "IN:1:__stream_0"
output_stream: "OUT:out"
}
)");
)pb");
EXPECT_THAT(graph.GetConfig(), EqualsProto(expected));
}
@@ -120,7 +120,7 @@ TEST(BuilderTest, TypedMultiple) {
adder.Out(MPP_TAG("OUT")).SetName("out") >> graph.Out("OUT");
CalculatorGraphConfig expected =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "IN:base"
output_stream: "OUT:out"
node {
@@ -134,7 +134,7 @@ TEST(BuilderTest, TypedMultiple) {
input_stream: "IN:1:__stream_0"
output_stream: "OUT:out"
}
)");
)pb");
EXPECT_THAT(graph.GetConfig(), EqualsProto(expected));
}
@@ -145,7 +145,7 @@ TEST(BuilderTest, PacketGenerator) {
generator.SideOut("OUT") >> graph.SideOut("OUT");
CalculatorGraphConfig expected =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_side_packet: "IN:__side_packet_0"
output_side_packet: "OUT:__side_packet_1"
packet_generator {
@@ -153,7 +153,7 @@ TEST(BuilderTest, PacketGenerator) {
input_side_packet: "IN:__side_packet_0"
output_side_packet: "OUT:__side_packet_1"
}
)");
)pb");
EXPECT_THAT(graph.GetConfig(), EqualsProto(expected));
}
@@ -167,7 +167,7 @@ TEST(BuilderTest, EmptyTag) {
foo.Out("")[1].SetName("y") >> graph.Out("TWO");
CalculatorGraphConfig expected =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "A:a"
input_stream: "B:b"
input_stream: "C:c"
@@ -181,7 +181,7 @@ TEST(BuilderTest, EmptyTag) {
output_stream: "x"
output_stream: "y"
}
)");
)pb");
EXPECT_THAT(graph.GetConfig(), EqualsProto(expected));
}
mediapipe/framework/api2/node_test.cc
+24 -24
View File
@@ -132,12 +132,12 @@ TEST(NodeTest, GetContract) {
// with what you have in the graph, then you let the calculator fill it in
// with what it expects, and then you see if they match.
const CalculatorGraphConfig::Node node_config =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig::Node>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig::Node>(R"pb(
calculator: "Foo"
input_stream: "BASE:base"
input_stream: "SCALE:scale"
output_stream: "OUT:out"
)");
)pb");
mediapipe::CalculatorContract contract;
MP_EXPECT_OK(contract.Initialize(node_config));
MP_EXPECT_OK(Foo::Contract::GetContract(&contract));
@@ -147,13 +147,13 @@ TEST(NodeTest, GetContract) {
TEST(NodeTest, GetContractMulti) {
const CalculatorGraphConfig::Node node_config =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig::Node>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig::Node>(R"pb(
calculator: "Baz"
input_stream: "DATA:0:b"
input_stream: "DATA:1:c"
output_stream: "DATA:0:d"
output_stream: "DATA:1:e"
)");
)pb");
mediapipe::CalculatorContract contract;
MP_EXPECT_OK(contract.Initialize(node_config));
MP_EXPECT_OK(Baz::Contract::GetContract(&contract));
@@ -204,7 +204,7 @@ TEST(NodeTest, RunInGraph5) { RunFooCalculatorInGraph("Foo5"); }
TEST(NodeTest, OptionalStream) {
CalculatorGraphConfig config =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "base"
input_side_packet: "bias"
output_stream: "out"
@@ -214,7 +214,7 @@ TEST(NodeTest, OptionalStream) {
input_side_packet: "BIAS:bias"
output_stream: "OUT:out"
}
)");
)pb");
std::vector<mediapipe::Packet> out_packets;
tool::AddVectorSink("out", &config, &out_packets);
mediapipe::CalculatorGraph graph;
@@ -229,7 +229,7 @@ TEST(NodeTest, OptionalStream) {
TEST(NodeTest, DynamicTypes) {
CalculatorGraphConfig config =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "in"
output_stream: "out"
node {
@@ -242,7 +242,7 @@ TEST(NodeTest, DynamicTypes) {
input_stream: "IN:bar"
output_stream: "OUT:out"
}
)");
)pb");
std::vector<mediapipe::Packet> out_packets;
tool::AddVectorSink("out", &config, &out_packets);
mediapipe::CalculatorGraph graph;
@@ -257,7 +257,7 @@ TEST(NodeTest, DynamicTypes) {
TEST(NodeTest, MultiPort) {
CalculatorGraphConfig config =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "in0"
input_stream: "in1"
output_stream: "out0"
@@ -279,7 +279,7 @@ TEST(NodeTest, MultiPort) {
input_stream: "IN:baz1"
output_stream: "OUT:out1"
}
)");
)pb");
std::vector<mediapipe::Packet> out0_packets;
std::vector<mediapipe::Packet> out1_packets;
tool::AddVectorSink("out0", &config, &out0_packets);
@@ -325,7 +325,7 @@ MEDIAPIPE_REGISTER_NODE(SideFallback);
TEST(NodeTest, SideFallbackWithStream) {
CalculatorGraphConfig config =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "in"
input_stream: "factor"
output_stream: "out"
@@ -335,7 +335,7 @@ TEST(NodeTest, SideFallbackWithStream) {
input_stream: "FACTOR:factor"
output_stream: "OUT:out"
}
)");
)pb");
std::vector<int> outputs;
mediapipe::CalculatorGraph graph;
MP_EXPECT_OK(graph.Initialize(config, {}));
@@ -356,7 +356,7 @@ TEST(NodeTest, SideFallbackWithStream) {
TEST(NodeTest, SideFallbackWithSide) {
CalculatorGraphConfig config =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "in"
input_side_packet: "factor"
output_stream: "out"
@@ -366,7 +366,7 @@ TEST(NodeTest, SideFallbackWithSide) {
input_side_packet: "FACTOR:factor"
output_stream: "OUT:out"
}
)");
)pb");
std::vector<int> outputs;
mediapipe::CalculatorGraph graph;
MP_EXPECT_OK(graph.Initialize(config, {}));
@@ -385,7 +385,7 @@ TEST(NodeTest, SideFallbackWithSide) {
TEST(NodeTest, SideFallbackWithNone) {
CalculatorGraphConfig config =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "in"
output_stream: "out"
node {
@@ -393,7 +393,7 @@ TEST(NodeTest, SideFallbackWithNone) {
input_stream: "IN:in"
output_stream: "OUT:out"
}
)");
)pb");
std::vector<int> outputs;
mediapipe::CalculatorGraph graph;
auto status = graph.Initialize(config, {});
@@ -402,7 +402,7 @@ TEST(NodeTest, SideFallbackWithNone) {
TEST(NodeTest, SideFallbackWithBoth) {
CalculatorGraphConfig config =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "in"
input_stream: "factor"
input_side_packet: "factor_side"
@@ -414,7 +414,7 @@ TEST(NodeTest, SideFallbackWithBoth) {
input_side_packet: "FACTOR:factor_side"
output_stream: "OUT:out"
}
)");
)pb");
std::vector<int> outputs;
mediapipe::CalculatorGraph graph;
auto status = graph.Initialize(config, {});
@@ -423,7 +423,7 @@ TEST(NodeTest, SideFallbackWithBoth) {
TEST(NodeTest, OneOf) {
CalculatorGraphConfig config =
::mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
::mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "in"
output_stream: "out"
node {
@@ -431,7 +431,7 @@ TEST(NodeTest, OneOf) {
input_stream: "IN:in"
output_stream: "OUT:out"
}
)");
)pb");
std::vector<mediapipe::Packet> out_packets;
tool::AddVectorSink("out", &config, &out_packets);
mediapipe::CalculatorGraph graph;
@@ -484,7 +484,7 @@ MEDIAPIPE_REGISTER_NODE(ListIntPackets);
TEST(NodeTest, DefaultTimestampChange0) {
CalculatorGraphConfig config =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "a"
input_stream: "b"
output_stream: "out"
@@ -504,7 +504,7 @@ TEST(NodeTest, DefaultTimestampChange0) {
input_stream: "INT:1:b"
output_stream: "STR:out"
}
)");
)pb");
std::vector<mediapipe::Packet> out_packets;
tool::AddVectorSink("out", &config, &out_packets);
mediapipe::CalculatorGraph graph;
@@ -541,7 +541,7 @@ MEDIAPIPE_REGISTER_NODE(ConsumerNode);
TEST(NodeTest, ConsumeInputs) {
CalculatorGraphConfig config =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "int"
input_stream: "any"
input_stream: "num"
@@ -551,7 +551,7 @@ TEST(NodeTest, ConsumeInputs) {
input_stream: "ANY:any"
input_stream: "NUM:num"
}
)");
)pb");
mediapipe::CalculatorGraph graph;
MP_EXPECT_OK(graph.Initialize(config, {}));
MP_EXPECT_OK(graph.StartRun({}));
mediapipe/framework/api2/subgraph_test.cc
+10 -10
View File
@@ -46,7 +46,7 @@ class FooBarImpl2 : public SubgraphImpl<FooBar2, FooBarImpl2> {
TEST(SubgraphTest, SubgraphConfig) {
CalculatorGraphConfig subgraph = FooBarImpl1().GetConfig({}).value();
const CalculatorGraphConfig expected_graph =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "IN:__stream_0"
output_stream: "OUT:__stream_2"
node {
@@ -59,14 +59,14 @@ TEST(SubgraphTest, SubgraphConfig) {
input_stream: "IN:__stream_1"
output_stream: "OUT:__stream_2"
}
)");
)pb");
EXPECT_THAT(subgraph, EqualsProto(expected_graph));
}
TEST(SubgraphTest, TypedSubgraphConfig) {
CalculatorGraphConfig subgraph = FooBarImpl2().GetConfig({}).value();
const CalculatorGraphConfig expected_graph =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "IN:__stream_0"
output_stream: "OUT:__stream_2"
node {
@@ -79,7 +79,7 @@ TEST(SubgraphTest, TypedSubgraphConfig) {
input_stream: "IN:__stream_1"
output_stream: "OUT:__stream_2"
}
)");
)pb");
EXPECT_THAT(subgraph, EqualsProto(expected_graph));
}
@@ -97,7 +97,7 @@ TEST(SubgraphTest, ProtoApiConfig) {
bar->add_output_stream("OUT:__stream_2");
const CalculatorGraphConfig expected_graph =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
input_stream: "IN:__stream_0"
output_stream: "OUT:__stream_2"
node {
@@ -110,13 +110,13 @@ TEST(SubgraphTest, ProtoApiConfig) {
input_stream: "IN:__stream_1"
output_stream: "OUT:__stream_2"
}
)");
)pb");
EXPECT_THAT(graph, EqualsProto(expected_graph));
}
TEST(SubgraphTest, ExpandSubgraphs) {
CalculatorGraphConfig supergraph =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
node {
name: "simple_source"
calculator: "SomeSourceCalculator"
@@ -127,9 +127,9 @@ TEST(SubgraphTest, ExpandSubgraphs) {
input_stream: "IN:foo"
output_stream: "OUT:output"
}
)");
)pb");
const CalculatorGraphConfig expected_graph =
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"(
mediapipe::ParseTextProtoOrDie<CalculatorGraphConfig>(R"pb(
node {
name: "simple_source"
calculator: "SomeSourceCalculator"
@@ -147,7 +147,7 @@ TEST(SubgraphTest, ExpandSubgraphs) {
input_stream: "IN:foobar____stream_1"
output_stream: "OUT:output"
}
)");
)pb");
MP_EXPECT_OK(tool::ExpandSubgraphs(&supergraph));
EXPECT_THAT(supergraph, EqualsProto(expected_graph));
}