Renamed NNTensor

themarpe · themarpe · commit 2454df304714 · 2020-10-13T22:23:02.000+02:00
diff --git a/depthai-core b/depthai-core
@@ -1 +1 @@
-Subproject commit 050169c29f761fe7d79500a0eb535d72c52b2637
+Subproject commit 0f22d73940ce666780065151cab8ba51fba28cda
diff --git a/src/DatatypeBindings.cpp b/src/DatatypeBindings.cpp
@@ -6,7 +6,7 @@
 // depthai-shared
 #include "depthai-shared/datatype/RawBuffer.hpp"
 #include "depthai-shared/datatype/ImgFrame.hpp"
-#include "depthai-shared/datatype/NNTensor.hpp"
+#include "depthai-shared/datatype/NNData.hpp"
 
 
 //pybind
@@ -112,11 +112,11 @@ void DatatypeBindings::bind(pybind11::module& m){
 
 
 
-    // NNTensor
-    py::class_<NNTensor, RawBuffer, std::shared_ptr<NNTensor>> nnTensor(m, "NNTensor");
-    nnTensor
-        .def_readwrite("tensors", &NNTensor::tensors)
-        .def_readwrite("batchSize", &NNTensor::batchSize)
+    // NNData
+    py::class_<NNData, RawBuffer, std::shared_ptr<NNData>> nnData(m, "NNData");
+    nnData
+        .def_readwrite("tensors", &NNData::tensors)
+        .def_readwrite("batchSize", &NNData::batchSize)
         ;
 
     py::class_<TensorInfo> tensorInfo(m, "TensorInfo");
@@ -136,7 +136,6 @@ void DatatypeBindings::bind(pybind11::module& m){
         .value("INT", TensorInfo::DataType::INT)
         .value("FP32", TensorInfo::DataType::FP32)
         .value("I8", TensorInfo::DataType::I8)
-        .export_values()
         ;
         
     py::enum_<TensorInfo::StorageOrder>(tensorInfo, "StorageOrder")
@@ -154,7 +153,6 @@ void DatatypeBindings::bind(pybind11::module& m){
         .value("C", TensorInfo::StorageOrder::C)
         .value("H", TensorInfo::StorageOrder::H)
         .value("W", TensorInfo::StorageOrder::W)
-        .export_values()
         ;
 
 
diff --git a/src/py_bindings.cpp b/src/py_bindings.cpp
@@ -8,6 +8,7 @@
 #include <vector>
 
 // pybind11
+#include "pybind11_common.hpp"
 
 // depthai-core
 #include "depthai/Device.hpp"
@@ -19,6 +20,7 @@
 //#include "py_tensor_entry_container_iterator.hpp"
 //#include "device_bindings.hpp"
 
+
 #include "pipeline/PipelineBindings.hpp"
 #include "pipeline/NodeBindings.hpp"
 #include "XLinkConnectionBindings.hpp"
