rknncli 0.2.0__py3-none-any.whl

rknncli/schema/rknn/Type1.py

@@ -0,0 +1,94 @@
+# automatically generated by the FlatBuffers compiler, do not modify
+
+# namespace: rknn
+
+import flatbuffers
+from flatbuffers.compat import import_numpy
+np = import_numpy()
+
+class Type1(object):
+    __slots__ = ['_tab']
+
+    @classmethod
+    def GetRootAs(cls, buf, offset=0):
+        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
+        x = Type1()
+        x.Init(buf, n + offset)
+        return x
+
+    @classmethod
+    def GetRootAsType1(cls, buf, offset=0):
+        """This method is deprecated. Please switch to GetRootAs."""
+        return cls.GetRootAs(buf, offset)
+    @classmethod
+    def Type1BufferHasIdentifier(cls, buf, offset, size_prefixed=False):
+        return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x52\x4B\x4E\x4E", size_prefixed=size_prefixed)
+
+    # Type1
+    def Init(self, buf, pos):
+        self._tab = flatbuffers.table.Table(buf, pos)
+
+    # Type1
+    def Var1(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
+        if o != 0:
+            return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
+        return 0
+
+def Type1Start(builder):
+    builder.StartObject(1)
+
+def Start(builder):
+    Type1Start(builder)
+
+def Type1AddVar1(builder, var1):
+    builder.PrependInt32Slot(0, var1, 0)
+
+def AddVar1(builder, var1):
+    Type1AddVar1(builder, var1)
+
+def Type1End(builder):
+    return builder.EndObject()
+
+def End(builder):
+    return Type1End(builder)
+
+
+class Type1T(object):
+
+    # Type1T
+    def __init__(
+        self,
+        var1 = 0,
+    ):
+        self.var1 = var1  # type: int
+
+    @classmethod
+    def InitFromBuf(cls, buf, pos):
+        type1 = Type1()
+        type1.Init(buf, pos)
+        return cls.InitFromObj(type1)
+
+    @classmethod
+    def InitFromPackedBuf(cls, buf, pos=0):
+        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
+        return cls.InitFromBuf(buf, pos+n)
+
+    @classmethod
+    def InitFromObj(cls, type1):
+        x = Type1T()
+        x._UnPack(type1)
+        return x
+
+    # Type1T
+    def _UnPack(self, type1):
+        if type1 is None:
+            return
+        self.var1 = type1.Var1()
+
+    # Type1T
+    def Pack(self, builder):
+        Type1Start(builder)
+        Type1AddVar1(builder, self.var1)
+        type1 = Type1End(builder)
+        return type1

rknncli/schema/rknn/Type2.py

@@ -0,0 +1,255 @@
+# automatically generated by the FlatBuffers compiler, do not modify
+
+# namespace: rknn
+
+import flatbuffers
+from flatbuffers.compat import import_numpy
+np = import_numpy()
+
+class Type2(object):
+    __slots__ = ['_tab']
+
+    @classmethod
+    def GetRootAs(cls, buf, offset=0):
+        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
+        x = Type2()
+        x.Init(buf, n + offset)
+        return x
+
+    @classmethod
+    def GetRootAsType2(cls, buf, offset=0):
+        """This method is deprecated. Please switch to GetRootAs."""
+        return cls.GetRootAs(buf, offset)
+    @classmethod
+    def Type2BufferHasIdentifier(cls, buf, offset, size_prefixed=False):
+        return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x52\x4B\x4E\x4E", size_prefixed=size_prefixed)
+
+    # Type2
+    def Init(self, buf, pos):
+        self._tab = flatbuffers.table.Table(buf, pos)
+
+    # Type2
+    def Var1(self, j):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
+        if o != 0:
+            a = self._tab.Vector(o)
+            return self._tab.Get(flatbuffers.number_types.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
+        return 0
+
+    # Type2
+    def Var1AsNumpy(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
+        if o != 0:
+            return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
+        return 0
+
+    # Type2
+    def Var1Length(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
+        if o != 0:
+            return self._tab.VectorLen(o)
+        return 0
+
+    # Type2
+    def Var1IsNone(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
+        return o == 0
+
+    # Type2
+    def Var2(self, j):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
+        if o != 0:
+            a = self._tab.Vector(o)
+            return self._tab.Get(flatbuffers.number_types.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
+        return 0
+
+    # Type2
+    def Var2AsNumpy(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
+        if o != 0:
+            return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
+        return 0
+
+    # Type2
+    def Var2Length(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
+        if o != 0:
+            return self._tab.VectorLen(o)
+        return 0
+
+    # Type2
+    def Var2IsNone(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(6))
+        return o == 0
+
+    # Type2
+    def Var3(self, j):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
+        if o != 0:
+            a = self._tab.Vector(o)
+            return self._tab.Get(flatbuffers.number_types.Int32Flags, a + flatbuffers.number_types.UOffsetTFlags.py_type(j * 4))
+        return 0
+
+    # Type2
+    def Var3AsNumpy(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
+        if o != 0:
+            return self._tab.GetVectorAsNumpy(flatbuffers.number_types.Int32Flags, o)
+        return 0
+
+    # Type2
+    def Var3Length(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
+        if o != 0:
+            return self._tab.VectorLen(o)
+        return 0
+
+    # Type2
+    def Var3IsNone(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(8))
+        return o == 0
+
+def Type2Start(builder):
+    builder.StartObject(3)
+
+def Start(builder):
+    Type2Start(builder)
+
+def Type2AddVar1(builder, var1):
+    builder.PrependUOffsetTRelativeSlot(0, flatbuffers.number_types.UOffsetTFlags.py_type(var1), 0)
+
+def AddVar1(builder, var1):
+    Type2AddVar1(builder, var1)
+
+def Type2StartVar1Vector(builder, numElems):
+    return builder.StartVector(4, numElems, 4)
+
+def StartVar1Vector(builder, numElems):
+    return Type2StartVar1Vector(builder, numElems)
+
+def Type2AddVar2(builder, var2):
+    builder.PrependUOffsetTRelativeSlot(1, flatbuffers.number_types.UOffsetTFlags.py_type(var2), 0)
+
+def AddVar2(builder, var2):
+    Type2AddVar2(builder, var2)
+
+def Type2StartVar2Vector(builder, numElems):
+    return builder.StartVector(4, numElems, 4)
+
+def StartVar2Vector(builder, numElems):
+    return Type2StartVar2Vector(builder, numElems)
+
+def Type2AddVar3(builder, var3):
+    builder.PrependUOffsetTRelativeSlot(2, flatbuffers.number_types.UOffsetTFlags.py_type(var3), 0)
+
+def AddVar3(builder, var3):
+    Type2AddVar3(builder, var3)
+
+def Type2StartVar3Vector(builder, numElems):
+    return builder.StartVector(4, numElems, 4)
+
+def StartVar3Vector(builder, numElems):
+    return Type2StartVar3Vector(builder, numElems)
+
+def Type2End(builder):
+    return builder.EndObject()
+
+def End(builder):
+    return Type2End(builder)
+
+try:
+    from typing import List
+except:
+    pass
+
+class Type2T(object):
+
+    # Type2T
+    def __init__(
+        self,
+        var1 = None,
+        var2 = None,
+        var3 = None,
+    ):
+        self.var1 = var1  # type: Optional[List[int]]
+        self.var2 = var2  # type: Optional[List[int]]
+        self.var3 = var3  # type: Optional[List[int]]
+
+    @classmethod
+    def InitFromBuf(cls, buf, pos):
+        type2 = Type2()
+        type2.Init(buf, pos)
+        return cls.InitFromObj(type2)
+
+    @classmethod
+    def InitFromPackedBuf(cls, buf, pos=0):
+        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
+        return cls.InitFromBuf(buf, pos+n)
+
+    @classmethod
+    def InitFromObj(cls, type2):
+        x = Type2T()
+        x._UnPack(type2)
+        return x
+
+    # Type2T
+    def _UnPack(self, type2):
+        if type2 is None:
+            return
+        if not type2.Var1IsNone():
+            if np is None:
+                self.var1 = []
+                for i in range(type2.Var1Length()):
+                    self.var1.append(type2.Var1(i))
+            else:
+                self.var1 = type2.Var1AsNumpy()
+        if not type2.Var2IsNone():
+            if np is None:
+                self.var2 = []
+                for i in range(type2.Var2Length()):
+                    self.var2.append(type2.Var2(i))
+            else:
+                self.var2 = type2.Var2AsNumpy()
+        if not type2.Var3IsNone():
+            if np is None:
+                self.var3 = []
+                for i in range(type2.Var3Length()):
+                    self.var3.append(type2.Var3(i))
+            else:
+                self.var3 = type2.Var3AsNumpy()
+
+    # Type2T
+    def Pack(self, builder):
+        if self.var1 is not None:
+            if np is not None and type(self.var1) is np.ndarray:
+                var1 = builder.CreateNumpyVector(self.var1)
+            else:
+                Type2StartVar1Vector(builder, len(self.var1))
+                for i in reversed(range(len(self.var1))):
+                    builder.PrependInt32(self.var1[i])
+                var1 = builder.EndVector()
+        if self.var2 is not None:
+            if np is not None and type(self.var2) is np.ndarray:
+                var2 = builder.CreateNumpyVector(self.var2)
+            else:
+                Type2StartVar2Vector(builder, len(self.var2))
+                for i in reversed(range(len(self.var2))):
+                    builder.PrependInt32(self.var2[i])
+                var2 = builder.EndVector()
+        if self.var3 is not None:
+            if np is not None and type(self.var3) is np.ndarray:
+                var3 = builder.CreateNumpyVector(self.var3)
+            else:
+                Type2StartVar3Vector(builder, len(self.var3))
+                for i in reversed(range(len(self.var3))):
+                    builder.PrependInt32(self.var3[i])
+                var3 = builder.EndVector()
+        Type2Start(builder)
+        if self.var1 is not None:
+            Type2AddVar1(builder, var1)
+        if self.var2 is not None:
+            Type2AddVar2(builder, var2)
+        if self.var3 is not None:
+            Type2AddVar3(builder, var3)
+        type2 = Type2End(builder)
+        return type2

rknncli/schema/rknn/Type3.py

@@ -0,0 +1,94 @@
+# automatically generated by the FlatBuffers compiler, do not modify
+
+# namespace: rknn
+
+import flatbuffers
+from flatbuffers.compat import import_numpy
+np = import_numpy()
+
+class Type3(object):
+    __slots__ = ['_tab']
+
+    @classmethod
+    def GetRootAs(cls, buf, offset=0):
+        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, offset)
+        x = Type3()
+        x.Init(buf, n + offset)
+        return x
+
+    @classmethod
+    def GetRootAsType3(cls, buf, offset=0):
+        """This method is deprecated. Please switch to GetRootAs."""
+        return cls.GetRootAs(buf, offset)
+    @classmethod
+    def Type3BufferHasIdentifier(cls, buf, offset, size_prefixed=False):
+        return flatbuffers.util.BufferHasIdentifier(buf, offset, b"\x52\x4B\x4E\x4E", size_prefixed=size_prefixed)
+
+    # Type3
+    def Init(self, buf, pos):
+        self._tab = flatbuffers.table.Table(buf, pos)
+
+    # Type3
+    def Var1(self):
+        o = flatbuffers.number_types.UOffsetTFlags.py_type(self._tab.Offset(4))
+        if o != 0:
+            return self._tab.Get(flatbuffers.number_types.Int32Flags, o + self._tab.Pos)
+        return 0
+
+def Type3Start(builder):
+    builder.StartObject(1)
+
+def Start(builder):
+    Type3Start(builder)
+
+def Type3AddVar1(builder, var1):
+    builder.PrependInt32Slot(0, var1, 0)
+
+def AddVar1(builder, var1):
+    Type3AddVar1(builder, var1)
+
+def Type3End(builder):
+    return builder.EndObject()
+
+def End(builder):
+    return Type3End(builder)
+
+
+class Type3T(object):
+
+    # Type3T
+    def __init__(
+        self,
+        var1 = 0,
+    ):
+        self.var1 = var1  # type: int
+
+    @classmethod
+    def InitFromBuf(cls, buf, pos):
+        type3 = Type3()
+        type3.Init(buf, pos)
+        return cls.InitFromObj(type3)
+
+    @classmethod
+    def InitFromPackedBuf(cls, buf, pos=0):
+        n = flatbuffers.encode.Get(flatbuffers.packer.uoffset, buf, pos)
+        return cls.InitFromBuf(buf, pos+n)
+
+    @classmethod
+    def InitFromObj(cls, type3):
+        x = Type3T()
+        x._UnPack(type3)
+        return x
+
+    # Type3T
+    def _UnPack(self, type3):
+        if type3 is None:
+            return
+        self.var1 = type3.Var1()
+
+    # Type3T
+    def Pack(self, builder):
+        Type3Start(builder)
+        Type3AddVar1(builder, self.var1)
+        type3 = Type3End(builder)
+        return type3

rknncli-0.2.0.dist-info/METADATA

@@ -0,0 +1,69 @@
+Metadata-Version: 2.4
+Name: rknncli
+Version: 0.2.0
+Summary: A command line tool for parsing and displaying RKNN model information
+Author: rknncli
+License: MIT
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+
+# RKNN CLI
+
+A command line tool for parsing and displaying RKNN model information.
+
+## Installation
+
+```bash
+pip install -e .
+```
+
+## Usage
+
+```bash
+rknncli <path-to-rknn-model>
+```
+
+Example:
+
+```bash
+rknncli assets/yolov5s-640-640.rknn
+```
+
+## Output Format
+
+The tool prints model information in the following format:
+
+```
+Model: rknn model
+Version: 1.6.2-source_code
+Target Platform: rk3588
+
+Input information
+--------------------------------------------------------------------------------
+  ValueInfo "images": type INT8, shape ['batch', 3, 'height', 'width'],
+
+Output information
+--------------------------------------------------------------------------------
+  ValueInfo "output0": type INT8, shape ['batch', 255, 80, 80],
+  ValueInfo "286": type INT8, shape ['batch', 255, 40, 40],
+  ValueInfo "288": type INT8, shape ['batch', 255, 20, 20'],
+```
+
+## Development
+
+```bash
+# Install in development mode
+pip install -e .
+
+# Run the CLI
+python -m rknncli.cli assets/yolov5s-640-640.rknn
+```

rknncli-0.2.0.dist-info/RECORD

@@ -0,0 +1,16 @@
+rknncli/__init__.py,sha256=_xwmD7Q3XeNRSKVea1V9Yf5ZrFUdE9TRk09NOw87zTQ,111
+rknncli/cli.py,sha256=xsvGye9fsaRO3vDoN6FBYLdityibR5_SsRhNGFuJgX8,6831
+rknncli/parser.py,sha256=45CrVa18i2AOZ_UseyQQp6xySQ70GMYpmDh3P3Us_iU,11987
+rknncli/schema/rknn/Graph.py,sha256=RgRgauVQ8agqx0VCfNBl-VAZPq4qBRgcbzYKYOc-Qgs,12591
+rknncli/schema/rknn/Model.py,sha256=3RubV822E-Q9DNfOvZUwi9LSheV88XlzlJNdQfj7NWE,13797
+rknncli/schema/rknn/Node.py,sha256=jF8ZKXdQRZqATBhqffJLoY7ETMN0KF5bpaTii8J66WI,12052
+rknncli/schema/rknn/Tensor.py,sha256=QTmyIITukptbA9rGcHBCaBingEotevh7ev7ZD4IC9Mo,23476
+rknncli/schema/rknn/Type1.py,sha256=9lRda2jNpvgm8yVWraNCEhv_qCFkmgqCsxMNSOa9zXc,2330
+rknncli/schema/rknn/Type2.py,sha256=Hz2waIXoPJ3npsSE3yoNN6y0nss2_uu7JCoccPg83hw,8128
+rknncli/schema/rknn/Type3.py,sha256=9uV1-WF2jhMJLFzo2k7MQdhDnif2lsd2J1C5enskyPc,2330
+rknncli/schema/rknn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rknncli-0.2.0.dist-info/METADATA,sha256=mC_vsPRsRi6y1w-c1wF9_yElVnUB2s-HUL_QSRxXE-4,1659
+rknncli-0.2.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+rknncli-0.2.0.dist-info/entry_points.txt,sha256=CiPnMUneLqwv2pcjg4V5sQbQDTdx-hi7t5W5glqYZVM,45
+rknncli-0.2.0.dist-info/top_level.txt,sha256=HPlOMKakzngvN2n2l6Bsg8N1Xflsm7fyCbPO6Fx_7iA,8
+rknncli-0.2.0.dist-info/RECORD,,

rknncli-0.2.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.10.2)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

rknncli-0.2.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+rknncli = rknncli.cli:main

rknncli-0.2.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+rknncli