diff --git a/deploy/cpp/src/main.cc b/deploy/cpp/src/main.cc index 00dd325680b39df3900689e98fae6f32b58ec99e..ca25a46e1cc132794faabc6235239b33433ba25b 100644 --- a/deploy/cpp/src/main.cc +++ b/deploy/cpp/src/main.cc @@ -29,6 +29,7 @@ #endif #include "include/object_detector.h" +#include DEFINE_string(model_dir, "", "Path of inference model"); diff --git a/static/deploy/cpp/src/main.cc b/static/deploy/cpp/src/main.cc index f2b4b1ad26b923df7f597f8ef88c9a4850bff23f..2e3b356c8cc09cb2d8075d0d823d1a4dc2d7ffb3 100644 --- a/static/deploy/cpp/src/main.cc +++ b/static/deploy/cpp/src/main.cc @@ -29,6 +29,7 @@ #endif #include "include/object_detector.h" +#include DEFINE_string(model_dir, "", "Path of inference model");