#define TENSOR_FILTER_TENSORFLOW_H
#ifdef __cplusplus
-#include <stdint.h>
#include <glib.h>
+#include <gst/gst.h>
#include <setjmp.h>
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <fstream>
#include <algorithm>
-#include <vector>
+#include <vector>
#include <tensorflow/cc/ops/const_op.h>
#include <tensorflow/cc/ops/image_ops.h>
#include <tensorflow/core/lib/strings/str_util.h>
#include <tensorflow/tools/graph_transforms/transform_utils.h>
-#include <tensor_common.h>
+#include <tensor_typedef.h>
using namespace tensorflow;
const char* getModelPath();
int setInputTensorProp ();
int setOutputTensorProp ();
-
+
int getInputTensorSize ();
int getOutputTensorSize ();
int getInputTensorDim (GstTensorsInfo * info);
* @bug No known bugs.
*/
-#include <sys/time.h>
#include <unistd.h>
#include <algorithm>
TFLiteCore::TFLiteCore (const char *_model_path)
{
model_path = _model_path;
-
- gst_tensors_info_init (&inputTensorMeta);
- gst_tensors_info_init (&outputTensorMeta);
}
/**
#ifdef __cplusplus
#include <iostream>
-#include <stdint.h>
#include <glib.h>
+#include <gst/gst.h>
#include <tensorflow/contrib/lite/model.h>
#include <tensorflow/contrib/lite/optional_debug_tools.h>
#include <tensorflow/contrib/lite/string_util.h>
#include <tensorflow/contrib/lite/kernels/register.h>
-#include <tensor_common.h>
+#include <tensor_typedef.h>
/**
* @brief ring cache structure