Add support for flatbuffers 2.0

Signed-off-by: Tom Wildenhain <[email protected]>

Author: TomWildenhain-Microsoft

tests/test_tflite_postprocess.py

@@ -22,6 +22,12 @@
 
 # pylint: disable=missing-docstring
 
+def endvector(builder, length):
+    try:
+        builder.EndVector(length)
+    except TypeError:
+        # flatbuffers 2.0 changes the API
+        builder.EndVector()
 
 class TFLiteDetectionPostProcessTests(Tf2OnnxBackendTestBase):
 
@@ -90,7 +96,7 @@ def make_postprocess_model(self, max_detections=10, detections_per_class=100, ma
         # op_codes
         Model.ModelStartOperatorCodesVector(builder, 1)
         builder.PrependUOffsetTRelative(op_code)
-        op_codes = builder.EndVector(1)
+        op_codes = endvector(builder, 1)
 
         # Make tensors
         # [names, shape, type tensors]
@@ -118,19 +124,19 @@ def make_postprocess_model(self, max_detections=10, detections_per_class=100, ma
         SubGraph.SubGraphStartTensorsVector(builder, len(ts))
         for tensor in reversed(ts):
             builder.PrependUOffsetTRelative(tensor)
-        tensors = builder.EndVector(len(ts))
+        tensors = endvector(builder, len(ts))
 
         # inputs
         SubGraph.SubGraphStartInputsVector(builder, 3)
         for inp in reversed([0, 1, 2]):
             builder.PrependInt32(inp)
-        inputs = builder.EndVector(3)
+        inputs = endvector(builder, 3)
 
         # outputs
         SubGraph.SubGraphStartOutputsVector(builder, 4)
         for out in reversed([3, 4, 5, 6]):
             builder.PrependInt32(out)
-        outputs = builder.EndVector(4)
+        outputs = endvector(builder, 4)
 
         flexbuffer = \
             b'y_scale\x00nms_score_threshold\x00max_detections\x00x_scale\x00w_scale\x00nms_iou_threshold' \
@@ -164,7 +170,7 @@ def make_postprocess_model(self, max_detections=10, detections_per_class=100, ma
         # operators
         SubGraph.SubGraphStartOperatorsVector(builder, 1)
         builder.PrependUOffsetTRelative(operator)
-        operators = builder.EndVector(1)
+        operators = endvector(builder, 1)
 
         # subgraph
         graph_name = builder.CreateString("TFLite graph")
@@ -179,20 +185,20 @@ def make_postprocess_model(self, max_detections=10, detections_per_class=100, ma
         # subgraphs
         Model.ModelStartSubgraphsVector(builder, 1)
         builder.PrependUOffsetTRelative(subgraph)
-        subgraphs = builder.EndVector(1)
+        subgraphs = endvector(builder, 1)
 
         description = builder.CreateString("Model for tflite testing")
 
         Buffer.BufferStartDataVector(builder, 0)
-        data = builder.EndVector(0)
+        data = endvector(builder, 0)
 
         Buffer.BufferStart(builder)
         Buffer.BufferAddData(builder, data)
         buffer = Buffer.BufferEnd(builder)
 
         Model.ModelStartBuffersVector(builder, 1)
         builder.PrependUOffsetTRelative(buffer)
-        buffers = builder.EndVector(1)
+        buffers = endvector(builder, 1)
 
         # model
         Model.ModelStart(builder)