Merge pull request #59 from luxonis/output_format_checks

NN output format checks

Author: SzabolcsGergely

depthai-core

@@ -214,6 +214,13 @@ static py::array* _getTensorPythonNumpyArray(unsigned char *data, TensorInfo ti)
 
 py::array* PyNNetPacket::getTensor(unsigned index)
 {
+    if(_NN_config[index].contains("output_format"))
+    {
+        if(_NN_config[index]["output_format"] != std::string("raw"))
+        {
+            throw std::runtime_error("getTensor should be used only when [\"NN_config\"][\"output_format\"] is set to raw! https://docs.luxonis.com/api/#creating-blob-configuration-file");
+        }
+    }
     assert(index < _tensors_info.size());
     TensorInfo ti = _tensors_info[index];
     unsigned char * data = _tensors_raw_data->data->data();

src/nnet_packet_bindings.cpp

@@ -20,6 +20,6 @@ struct PyNNetPacket : public NNetPacket {
 public:
     using NNetPacket::_tensors_raw_data;
     using NNetPacket::_tensor_name_to_index;
-
+    using NNetPacket::_NN_config;
 
 };