implements inference for the use of tensorflow_cc library.
Change-Id: Id63a0a0b04d9f429dc1b32a0278184d06be7c610
diff --git a/ForegroundSegmentation/pluginParameters.h b/ForegroundSegmentation/pluginParameters.h
index 5bd5d7d..ec38f22 100644
--- a/ForegroundSegmentation/pluginParameters.h
+++ b/ForegroundSegmentation/pluginParameters.h
@@ -5,11 +5,18 @@
#include <string>
#include <map>
-
struct PluginParameters {
std::string stream = "out";
+#ifdef TFLITE
+#ifdef __ANDROID
std::string model = "model_256_Qlatency.tflite";
- std::string image = "background1.png";
+#else
+ std::string model = "model_256_F_16.tflite";
+#endif
+#else
+ std::string model = "frozen_inference_graph.pb";
+#endif //TFLITE
+ std::string image = "background2.png";
};
void setGlobalPluginParameters(std::map<std::string, std::string> pp);