rknncli 0.2.0__tar.gz

rknncli-0.2.0/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,51 @@
+# 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/pyproject.toml

@@ -0,0 +1,47 @@
+[project]
+name = "rknncli"
+version = "0.2.0"
+description = "A command line tool for parsing and displaying RKNN model information"
+readme = "README.md"
+requires-python = ">=3.8"
+license = {text = "MIT"}
+authors = [
+    {name = "rknncli"},
+]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+]
+
+[project.scripts]
+rknncli = "rknncli.cli:main"
+
+[build-system]
+requires = ["setuptools>=45", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["rknncli*"]
+
+[tool.black]
+line-length = 100
+target-version = ['py38']
+
+[tool.ruff]
+line-length = 100
+select = ["E", "F", "I", "N", "W"]
+ignore = ["E501"]
+
+[tool.mypy]
+python_version = "3.8"
+warn_return_any = true
+warn_unused_configs = true
+disallow_untyped_defs = true

rknncli-0.2.0/rknncli/__init__.py

@@ -0,0 +1,3 @@
+"""RKNN CLI - A command line tool for parsing and displaying RKNN model information."""
+
+__version__ = "0.1.0"

rknncli-0.2.0/rknncli/cli.py

@@ -0,0 +1,229 @@
+"""RKNN CLI - Command line interface for parsing RKNN models."""
+
+import argparse
+import sys
+from pathlib import Path
+from typing import List, Dict, Any
+
+from rknncli.parser import RKNNParser
+
+
+def format_shape(size: List) -> List:
+    """Format tensor shape with dimension names.
+
+    Args:
+        size: List of dimension sizes.
+
+    Returns:
+        List of dimension names (strings) or dimension values.
+    """
+    return size
+
+
+def get_dtype_str(dtype_info: Dict) -> str:
+    """Get data type string from dtype info.
+
+    Args:
+        dtype_info: Dictionary containing dtype information.
+
+    Returns:
+        String representation of the data type.
+    """
+    if isinstance(dtype_info, dict):
+        # Try vx_type first, then qnt_type
+        vx_type = dtype_info.get("vx_type", "").strip()
+        if vx_type:
+            return vx_type.upper()
+        qnt_type = dtype_info.get("qnt_type", "").strip()
+        if qnt_type:
+            return qnt_type.upper()
+    return "FLOAT"
+
+
+
+
+def print_merged_model_info(parser: RKNNParser) -> None:
+    """Print merged model information from both FlatBuffers and JSON.
+
+    Args:
+        parser: RKNNParser instance with FlatBuffers support.
+    """
+    # Get basic model info from JSON
+    print(f"Model: {parser.get_model_name()}")
+    platforms = parser.get_target_platform()
+    if platforms:
+        print(f"Target Platform: {', '.join(platforms)}")
+
+    # Get FlatBuffers info
+    fb_info = parser.get_flatbuffers_info()
+    if fb_info:
+        print(f"Format: {fb_info.get('format', 'Unknown')}")
+        print(f"Source: {fb_info.get('source', 'Unknown')}")
+        print(f"Compiler: {fb_info.get('compiler', 'Unknown')}")
+        print(f"Runtime: {fb_info.get('runtime', 'Unknown')}")
+
+        if fb_info.get("num_graphs", 0) > 0:
+            print(f"Number of graphs: {fb_info['num_graphs']}")
+
+    print()
+
+
+def print_merged_io_info(parser: RKNNParser) -> None:
+    """Print merged input/output information from both FlatBuffers and JSON.
+
+    Args:
+        parser: RKNNParser instance with FlatBuffers support.
+    """
+    # Get merged IO info
+    inputs, outputs = parser.get_merged_io_info()
+
+    # Print input information
+    print("Input information")
+    print("-" * 80)
+
+    for tensor in inputs:
+        name = tensor.get("url", f"tensor_{tensor.get('tensor_id', 0)}")
+        dtype = tensor.get("dtype", {})
+
+        # Get dtype string
+        if isinstance(dtype, dict):
+            dtype_str = get_dtype_str(dtype)
+        else:
+            dtype_str = str(dtype).upper()
+
+        size = tensor.get("size", [])
+        shape = format_shape(size)
+        shape_str = "[" + ", ".join(f"'{s}'" if isinstance(s, str) else str(s) for s in shape) + "]"
+
+        # Build the base output string
+        output_parts = [f'  ValueInfo "{name}": type {dtype_str}, shape {shape_str}']
+
+        # Add layout info from FlatBuffers
+        if "layout" in tensor and tensor["layout"]:
+            layout = tensor["layout"].upper()
+            layout_ori = tensor.get("layout_ori", "").upper()
+            if layout_ori and layout_ori != layout:
+                output_parts.append(f"layout {layout}(ori:{layout_ori})")
+            else:
+                output_parts.append(f"layout {layout}")
+
+        # Add quantization info from FlatBuffers
+        if "quant_info" in tensor:
+            quant = tensor["quant_info"]
+            if quant.get("qmethod") or quant.get("qtype"):
+                output_parts.append(f"quant {quant['qmethod']} {quant['qtype']}")
+
+        # Print the merged info on one line
+        print(", ".join(output_parts) + ",")
+
+    print()
+
+    # Print output information
+    print("Output information")
+    print("-" * 80)
+
+    for tensor in outputs:
+        name = tensor.get("url", f"tensor_{tensor.get('tensor_id', 0)}")
+        dtype = tensor.get("dtype", {})
+
+        # Get dtype string
+        if isinstance(dtype, dict):
+            dtype_str = get_dtype_str(dtype)
+        else:
+            dtype_str = str(dtype).upper()
+
+        size = tensor.get("size", [])
+        shape = format_shape(size)
+        shape_str = "[" + ", ".join(f"'{s}'" if isinstance(s, str) else str(s) for s in shape) + "]"
+
+        # Build the base output string
+        output_parts = [f'  ValueInfo "{name}": type {dtype_str}, shape {shape_str}']
+
+        # Add layout info from FlatBuffers
+        if "layout" in tensor and tensor["layout"]:
+            layout = tensor["layout"].upper()
+            layout_ori = tensor.get("layout_ori", "").upper()
+            if layout_ori and layout_ori != layout:
+                output_parts.append(f"layout {layout}(ori:{layout_ori})")
+            else:
+                output_parts.append(f"layout {layout}")
+
+        # Add quantization info from FlatBuffers
+        if "quant_info" in tensor:
+            quant = tensor["quant_info"]
+            if quant.get("qmethod") or quant.get("qtype"):
+                output_parts.append(f"quant {quant['qmethod']} {quant['qtype']}")
+
+        # Print the merged info on one line
+        print(", ".join(output_parts) + ",")
+
+
+def print_model_summary(parser) -> None:
+    """Print model summary information.
+
+    Args:
+        parser: RKNNParser instance.
+    """
+    print(f"Model: {parser.get_model_name()}")
+    print(f"Version: {parser.get_version()}")
+    platforms = parser.get_target_platform()
+    if platforms:
+        print(f"Target Platform: {', '.join(platforms)}")
+    print()
+
+
+
+
+def main() -> int:
+    """Main entry point.
+
+    Returns:
+        Exit code (0 for success, non-zero for error).
+    """
+    parser = argparse.ArgumentParser(
+        prog="rknncli",
+        description="A command line tool for parsing and displaying RKNN model information.",
+    )
+    parser.add_argument(
+        "model",
+        type=str,
+        help="Path to the RKNN model file",
+    )
+    parser.add_argument(
+        "-v", "--version",
+        action="version",
+        version="%(prog)s 0.1.0",
+    )
+
+    args = parser.parse_args()
+
+    model_path = Path(args.model)
+    if not model_path.exists():
+        print(f"Error: File not found: {model_path}", file=sys.stderr)
+        return 1
+
+    if not model_path.is_file():
+        print(f"Error: Not a file: {model_path}", file=sys.stderr)
+        return 1
+
+    try:
+        # Always parse both FlatBuffers and JSON data
+        rknn_parser = RKNNParser(model_path, parse_flatbuffers=True)
+
+        # Print merged model information
+        print_merged_model_info(rknn_parser)
+
+        # Print merged IO information
+        print_merged_io_info(rknn_parser)
+    except ValueError as e:
+        print(f"Error: Failed to parse RKNN file: {e}", file=sys.stderr)
+        return 1
+    except Exception as e:
+        print(f"Error: {e}", file=sys.stderr)
+        return 1
+
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())