Add MocapNET

2026-10-18 07:41:03 +00:00 · 2023-12-15 04:22:41 +05:30
parent 8434144ded
commit 65e8934d5e
336 changed files with 100584 additions and 0 deletions
@@ -0,0 +1,50 @@
+#pragma once
+
+#include <vector>
+
+
+#ifdef USE_TENSORFLOW1
+  #include "../Tensorflow/tensorflow.hpp"
+#elif USE_TENSORFLOW2
+  #include "../Tensorflow2/tensorflow2.h"
+#endif
+
+struct neuralNetworkModel
+{
+#ifdef USE_TENSORFLOW1
+    struct TensorflowInstance model; 
+#elif USE_TENSORFLOW2
+    struct Tensorflow2Instance model; 
+#else
+    #warning "No tensorflow support signaled through a C define.."
+    void * model;
+#endif
+};
+
+
+void neuralNetworkPrintVersion();
+
+char * neuralNetworkGetPath(struct neuralNetworkModel * nn);
+
+int neuralNetworkLoad(
+                      struct neuralNetworkModel * nn,
+                      const char * filename,
+                      const char * inputTensor,
+                      const char * outputTensor,
+                      unsigned int numberOfInputElements,
+                      unsigned int numberOfOutputElements,
+                      unsigned int forceCPU
+                     );
+
+
+
+
+int neuralNetworkUnload(
+                        struct neuralNetworkModel * nn
+                       );
+
+
+std::vector<float> neuralNetworkExecute(
+                                         struct neuralNetworkModel * nn,
+                                         std::vector<float> input
+                                       );