yolo3 86 route error

2018-12-23 16:20:17 +01:00
parent 3bd725801d
commit 2e8d0b1002
4 changed files with 29 additions and 24 deletions
@@ -18,7 +18,9 @@ public:
    
    nvinfer1::ICudaEngine *engineRT;
    nvinfer1::IExecutionContext *contextRT;
-    void* buffersRT[2];
+
+    const static int MAX_BUFFERS_RT = 10;
+    void* buffersRT[MAX_BUFFERS_RT];
    int buf_input_idx, buf_output_idx;

    dataDim_t input_dim, output_dim;