onnxslim 0.1.79__tar.gz → 0.1.81__tar.gz

onnxslim-0.1.81/.codespellignore

@@ -0,0 +1 @@
+onnxslim/argparser.py

onnxslim-0.1.81/.github/FUNDING.yml

@@ -0,0 +1,15 @@
+# These are supported funding model platforms
+
+github: inisis
+patreon: # Replace with a single Patreon username
+open_collective: # Replace with a single Open Collective username
+ko_fi: # Replace with a single Ko-fi username
+tidelift: # Replace with a single Tidelift platform-name/package-name e.g., npm/babel
+community_bridge: # Replace with a single Community Bridge project-name e.g., cloud-foundry
+liberapay: # Replace with a single Liberapay username
+issuehunt: # Replace with a single IssueHunt username
+lfx_crowdfunding: # Replace with a single LFX Crowdfunding project-name e.g., cloud-foundry
+polar: # Replace with a single Polar username
+buy_me_a_coffee: # Replace with a single Buy Me a Coffee username
+thanks_dev: # Replace with a single thanks.dev username
+custom: # Replace with up to 4 custom sponsorship URLs e.g., ['link1', 'link2']

onnxslim-0.1.81/.github/workflows/ci.yml

@@ -0,0 +1,39 @@
+name: CI
+
+on:
+  push:
+    branches: ["main"]
+  pull_request:
+    branches: ["main"]
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  test:
+    runs-on: self-hosted
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
+
+    steps:
+      - uses: actions/checkout@v3
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v7
+        with:
+          enable-cache: true
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install dependencies
+        run: uv sync --dev
+
+      - name: Run all tests with coverage
+        run: uv run pytest --cov=onnxslim --cov-report=xml:coverage-ci.xml
+
+      - name: Upload coverage to Codecov
+        uses: codecov/codecov-action@v5
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}

onnxslim-0.1.81/.github/workflows/format.yml

@@ -0,0 +1,26 @@
+#  # Ultralytics 🚀 - AGPL-3.0 license
+#  # Ultralytics Actions https://github.com/ultralytics/actions
+#  # This workflow automatically formats code and documentation in PRs to official Ultralytics standards
+
+#  name: Ultralytics Actions
+
+#  on:
+#    push:
+#      branches: [main]
+#    pull_request:
+#      branches: [main]
+#      types: [opened, closed, synchronize]
+
+#  jobs:
+#    format:
+#      runs-on: ubuntu-latest
+#      steps:
+#        - name: Run Ultralytics Formatting
+#          uses: ultralytics/actions@main
+#          with:
+#            token: ${{ secrets.GITHUB_TOKEN }} # automatically generated, do not modify
+#            python: true # format Python code and docstrings
+#            markdown: true # format Markdown
+#            prettier: true # format YAML
+#            spelling: true # check spelling
+#            links: false # check broken links

onnxslim-0.1.81/.github/workflows/nightly-test.yml

@@ -0,0 +1,32 @@
+name: nightly-test
+
+on:
+  schedule:
+    - cron: "0 18 * * *" # Runs at 6:00 PM UTC every day, which is 2:00 AM Beijing Time the next day
+
+jobs:
+  build:
+    runs-on: self-hosted
+
+    steps:
+      - uses: actions/checkout@v3
+
+      - uses: actions/setup-python@v4
+        with:
+          python-version: "3.10"
+
+      - name: install dependency
+        run: |
+          python -m pip install --upgrade pip wheel setuptools
+          pip install .
+          pip install pytest onnxruntime
+
+      - name: benchmark test
+        run: |
+          python tests/test_benchmark.py
+
+      - name: model test
+        run: |
+          pip install .
+          pip install pytest pytest-xdist onnxruntime timm torchvision --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu
+          python tests/test_onnx_nets.py

onnxslim-0.1.81/.github/workflows/python-publish.yml

@@ -0,0 +1,38 @@
+# This workflow will upload a Python Package using Twine when a release is created
+# For more information see: https://docs.github.com/en/actions/automating-builds-and-tests/building-and-testing-python#publishing-to-package-registries
+
+# This workflow uses actions that are not certified by GitHub.
+# They are provided by a third-party and are governed by
+# separate terms of service, privacy policy, and support
+# documentation.
+
+name: Upload Python Package
+
+on:
+  release:
+    types: [published]
+
+permissions:
+  contents: read
+
+jobs:
+  deploy:
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python
+        uses: actions/setup-python@v3
+        with:
+          python-version: "3.x"
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip wheel setuptools
+          pip install build
+      - name: Build package
+        run: python -m build
+      - name: Publish package
+        uses: pypa/gh-action-pypi-publish@27b31702a0e7fc50959f5ad993c78deac1bdfc29
+        with:
+          user: __token__
+          password: ${{ secrets.PYPI_API_TOKEN }}

onnxslim-0.1.81/.gitignore

@@ -0,0 +1,133 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+env/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+*.egg-info/
+.installed.cfg
+*.egg
+tmp/
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+uv.lock
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*,cover
+.hypothesis/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# IPython Notebook
+.ipynb_checkpoints
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# dotenv
+.env
+
+# virtualenv
+venv/
+ENV/
+
+# Spyder project settings
+.spyderproject
+
+# Rope project settings
+.ropeproject
+
+# atom remote-sync package
+.remote-sync.json
+
+# weights
+weights/
+
+#DS_Store
+.DS_Store
+
+# dev stuff
+eval/
+eval.ipynb
+dev.ipynb
+.vscode/
+
+# not ready
+videos/
+templates/
+data/ssd_dataloader.py
+data/datasets/
+doc/visualize.py
+read_results.py
+ssd300_120000/
+demos/live
+webdemo.py
+test_data_aug.py
+
+# attributes
+
+# pycharm
+.idea/
+
+# temp checkout soln
+data/datasets/
+data/ssd_dataloader.py
+tests/model
+
+# pylint
+.pylintrc
+
+# onnx
+*.onnx

onnxslim-0.1.81/.prettierignore

@@ -0,0 +1 @@
+onnxslim/argparser.py

onnxslim-0.1.81/CODE_OF_CONDUCT.md

@@ -0,0 +1,128 @@
+# Contributor Covenant Code of Conduct
+
+## Our Pledge
+
+We as members, contributors, and leaders pledge to make participation in our
+community a harassment-free experience for everyone, regardless of age, body
+size, visible or invisible disability, ethnicity, sex characteristics, gender
+identity and expression, level of experience, education, socio-economic status,
+nationality, personal appearance, race, religion, or sexual identity
+and orientation.
+
+We pledge to act and interact in ways that contribute to an open, welcoming,
+diverse, inclusive, and healthy community.
+
+## Our Standards
+
+Examples of behavior that contributes to a positive environment for our
+community include:
+
+- Demonstrating empathy and kindness toward other people
+- Being respectful of differing opinions, viewpoints, and experiences
+- Giving and gracefully accepting constructive feedback
+- Accepting responsibility and apologizing to those affected by our mistakes,
+  and learning from the experience
+- Focusing on what is best not just for us as individuals, but for the
+  overall community
+
+Examples of unacceptable behavior include:
+
+- The use of sexualized language or imagery, and sexual attention or
+  advances of any kind
+- Trolling, insulting or derogatory comments, and personal or political attacks
+- Public or private harassment
+- Publishing others' private information, such as a physical or email
+  address, without their explicit permission
+- Other conduct which could reasonably be considered inappropriate in a
+  professional setting
+
+## Enforcement Responsibilities
+
+Community leaders are responsible for clarifying and enforcing our standards of
+acceptable behavior and will take appropriate and fair corrective action in
+response to any behavior that they deem inappropriate, threatening, offensive,
+or harmful.
+
+Community leaders have the right and responsibility to remove, edit, or reject
+comments, commits, code, wiki edits, issues, and other contributions that are
+not aligned to this Code of Conduct, and will communicate reasons for moderation
+decisions when appropriate.
+
+## Scope
+
+This Code of Conduct applies within all community spaces, and also applies when
+an individual is officially representing the community in public spaces.
+Examples of representing our community include using an official e-mail address,
+posting via an official social media account, or acting as an appointed
+representative at an online or offline event.
+
+## Enforcement
+
+Instances of abusive, harassing, or otherwise unacceptable behavior may be
+reported to the community leaders responsible for enforcement at
+desmond.yao@buaa.edu.cn.
+All complaints will be reviewed and investigated promptly and fairly.
+
+All community leaders are obligated to respect the privacy and security of the
+reporter of any incident.
+
+## Enforcement Guidelines
+
+Community leaders will follow these Community Impact Guidelines in determining
+the consequences for any action they deem in violation of this Code of Conduct:
+
+### 1. Correction
+
+**Community Impact**: Use of inappropriate language or other behavior deemed
+unprofessional or unwelcome in the community.
+
+**Consequence**: A private, written warning from community leaders, providing
+clarity around the nature of the violation and an explanation of why the
+behavior was inappropriate. A public apology may be requested.
+
+### 2. Warning
+
+**Community Impact**: A violation through a single incident or series
+of actions.
+
+**Consequence**: A warning with consequences for continued behavior. No
+interaction with the people involved, including unsolicited interaction with
+those enforcing the Code of Conduct, for a specified period of time. This
+includes avoiding interactions in community spaces as well as external channels
+like social media. Violating these terms may lead to a temporary or
+permanent ban.
+
+### 3. Temporary Ban
+
+**Community Impact**: A serious violation of community standards, including
+sustained inappropriate behavior.
+
+**Consequence**: A temporary ban from any sort of interaction or public
+communication with the community for a specified period of time. No public or
+private interaction with the people involved, including unsolicited interaction
+with those enforcing the Code of Conduct, is allowed during this period.
+Violating these terms may lead to a permanent ban.
+
+### 4. Permanent Ban
+
+**Community Impact**: Demonstrating a pattern of violation of community
+standards, including sustained inappropriate behavior, harassment of an
+individual, or aggression toward or disparagement of classes of individuals.
+
+**Consequence**: A permanent ban from any sort of public interaction within
+the community.
+
+## Attribution
+
+This Code of Conduct is adapted from the [Contributor Covenant][homepage],
+version 2.0, available at
+https://www.contributor-covenant.org/version/2/0/code_of_conduct.html.
+
+Community Impact Guidelines were inspired by [Mozilla's code of conduct
+enforcement ladder](https://github.com/mozilla/diversity).
+
+[homepage]: https://www.contributor-covenant.org
+
+For answers to common questions about this code of conduct, see the FAQ at
+https://www.contributor-covenant.org/faq. Translations are available at
+https://www.contributor-covenant.org/translations.

onnxslim-0.1.81/CONTRIBUTING.md

@@ -0,0 +1,86 @@
+# Contributing to OnnxSlim
+
+Thank you for your interest in contributing to OnnxSlim! This document provides guidelines and instructions to help you get started.
+
+## Code of Conduct
+
+Please read and follow our [Code of Conduct](CODE_OF_CONDUCT.md) to foster an open and welcoming environment.
+
+## How to Contribute
+
+### Reporting Bugs
+
+If you find a bug, please create an issue with the following information:
+
+- Clear description of the bug
+- Steps to reproduce
+- Expected behavior
+- Screenshots (if applicable)
+- Environment details (OS, Python version, ONNX version, etc.)
+
+### Suggesting Features
+
+We welcome feature suggestions! Please create an issue with:
+
+- Clear description of the feature
+- Rationale for the feature
+- Potential implementation approach (optional)
+
+### Pull Requests
+
+1. Fork the repository
+2. Create a new branch (`git checkout -b feature/amazing-feature`)
+3. Make your changes
+4. Run tests if available
+5. Commit your changes (`git commit -m 'Add amazing feature'`)
+6. Push to the branch (`git push origin feature/amazing-feature`)
+7. Open a Pull Request
+
+### Coding Standards
+
+- Follow the existing code style
+- Write clear, commented, and testable code
+- Keep commits focused and atomic
+- Write meaningful commit messages
+- Use type hints where appropriate
+
+## Development Setup
+
+1. Clone the repository
+2. Create a virtual environment
+3. Install dependencies
+
+```bash
+# Setup commands
+git clone https://github.com/username/OnnxSlim.git
+cd OnnxSlim
+pip install -e . # Install package in development mode
+```
+
+## Testing
+
+Please ensure your code passes all tests:
+
+```bash
+# Test commands
+pytest tests/test_onnxslim.py
+```
+
+## Documentation
+
+- Update documentation for any changed functionality
+- Document new features thoroughly
+- Use clear and concise language
+- Add docstrings to new functions and classes
+
+## Review Process
+
+1. Maintainers will review your PR
+2. Changes may be requested
+3. Once approved, your PR will be merged
+
+## License
+
+By contributing, you agree that your contributions will be licensed under the project's license.
+
+Thank you for contributing to OnnxSlim!

{onnxslim-0.1.79/onnxslim.egg-info → onnxslim-0.1.81}/PKG-INFO

@@ -1,38 +1,28 @@
 Metadata-Version: 2.4
 Name: onnxslim
-Version: 0.1.79
+Version: 0.1.81
 Summary: OnnxSlim: A Toolkit to Help Optimize Onnx Model
-Home-page: https://github.com/inisis/OnnxSlim
-Author: inisis
-Author-email: desmond.yao@buaa.edu.cn
-License: MIT
-Project-URL: Bug Tracker, https://github.com/inisis/OnnxSlim/issues
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
+Project-URL: homepage, https://github.com/inisis/OnnxSlim
+Project-URL: issues, https://github.com/inisis/OnnxSlim/issues
+Author-email: inisis <desmond.yao@buaa.edu.cn>
+License-Expression: MIT
+License-File: LICENSE
 Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Python: >=3.6
-Description-Content-Type: text/markdown
-License-File: LICENSE
+Requires-Python: >=3.8
+Requires-Dist: colorama
+Requires-Dist: ml-dtypes
 Requires-Dist: onnx
-Requires-Dist: sympy>=1.13.1
 Requires-Dist: packaging
-Requires-Dist: colorama
-Requires-Dist: ml_dtypes
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: license
-Dynamic: license-file
-Dynamic: project-url
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
-
-# OnnxSlim
+Requires-Dist: sympy>=1.13.1
+Description-Content-Type: text/markdown
+
+<p align="center">
+  <img src="assets/logo/onnxslim-banner-light.svg#gh-light-mode-only" alt="OnnxSlim Logo" width="640"/>
+  <img src="assets/logo/onnxslim-banner.svg#gh-dark-mode-only" alt="OnnxSlim Logo" width="640"/>
+</p>
 
 <p align="center">
     <a href="https://pypi.org/project/onnxslim">

{onnxslim-0.1.79 → onnxslim-0.1.81}/README.md

@@ -1,4 +1,7 @@
-# OnnxSlim
+<p align="center">
+  <img src="assets/logo/onnxslim-banner-light.svg#gh-light-mode-only" alt="OnnxSlim Logo" width="640"/>
+  <img src="assets/logo/onnxslim-banner.svg#gh-dark-mode-only" alt="OnnxSlim Logo" width="640"/>
+</p>
 
 <p align="center">
     <a href="https://pypi.org/project/onnxslim">

onnxslim-0.1.81/assets/logo/onnxslim-banner-light.svg

@@ -0,0 +1,47 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 1280 640" width="1280" height="640">
+  <defs>
+    <linearGradient id="oGradL" x1="0%" y1="0%" x2="100%" y2="100%">
+      <stop offset="0%" style="stop-color:#005CED;stop-opacity:1" />
+      <stop offset="100%" style="stop-color:#00B4D8;stop-opacity:1" />
+    </linearGradient>
+    <linearGradient id="sGradL" x1="0%" y1="0%" x2="0%" y2="100%">
+      <stop offset="0%" style="stop-color:#00D4AA;stop-opacity:1" />
+      <stop offset="50%" style="stop-color:#00C9B7;stop-opacity:1" />
+      <stop offset="100%" style="stop-color:#00B4D8;stop-opacity:1" />
+    </linearGradient>
+  </defs>
+
+  <!-- Background -->
+  <rect width="1280" height="640" fill="#FFFFFF"/>
+
+  <!-- Decorative background elements -->
+  <circle cx="100" cy="100" r="200" fill="#005CED" opacity="0.03"/>
+  <circle cx="1180" cy="540" r="250" fill="#00D4AA" opacity="0.05"/>
+
+  <!-- Logo group - smaller, positioned on left -->
+  <g transform="translate(120, 170)">
+    <!-- O - outer ring -->
+    <circle cx="150" cy="150" r="125" fill="none" stroke="url(#oGradL)" stroke-width="20"/>
+
+    <!-- Smooth S curve -->
+    <path d="M 172 80
+             Q 117 80, 112 117
+             Q 108 150, 150 150
+             Q 192 150, 188 188
+             Q 183 220, 126 220"
+          fill="none"
+          stroke="url(#sGradL)"
+          stroke-width="12"
+          stroke-linecap="round"/>
+
+    <!-- Nodes at key positions -->
+    <circle cx="172" cy="80" r="18" fill="#00D4AA"/>
+    <circle cx="112" cy="117" r="14" fill="#00CCBB"/>
+    <circle cx="150" cy="150" r="18" fill="#00C4C0"/>
+    <circle cx="188" cy="188" r="14" fill="#00BCC5"/>
+    <circle cx="126" cy="220" r="18" fill="#00B4D8"/>
+  </g>
+
+  <!-- Text "OnnxSlim" - larger, centered vertically -->
+  <text x="420" y="360" font-family="Arial, Helvetica, sans-serif" font-size="150" font-weight="bold" fill="#005CED">OnnxSlim</text>
+</svg>

onnxslim-0.1.81/assets/logo/onnxslim-banner.svg

@@ -0,0 +1,55 @@
+<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 1280 640" width="1280" height="640">
+  <defs>
+    <linearGradient id="bgGrad" x1="0%" y1="0%" x2="100%" y2="100%">
+      <stop offset="0%" style="stop-color:#0a1628;stop-opacity:1" />
+      <stop offset="100%" style="stop-color:#1a2d4a;stop-opacity:1" />
+    </linearGradient>
+    <linearGradient id="oGrad" x1="0%" y1="0%" x2="100%" y2="100%">
+      <stop offset="0%" style="stop-color:#005CED;stop-opacity:1" />
+      <stop offset="100%" style="stop-color:#00B4D8;stop-opacity:1" />
+    </linearGradient>
+    <linearGradient id="sGrad" x1="0%" y1="0%" x2="0%" y2="100%">
+      <stop offset="0%" style="stop-color:#00D4AA;stop-opacity:1" />
+      <stop offset="50%" style="stop-color:#00C9B7;stop-opacity:1" />
+      <stop offset="100%" style="stop-color:#00B4D8;stop-opacity:1" />
+    </linearGradient>
+    <linearGradient id="textGrad" x1="0%" y1="0%" x2="100%" y2="0%">
+      <stop offset="0%" style="stop-color:#FFFFFF;stop-opacity:1" />
+      <stop offset="100%" style="stop-color:#E0E0E0;stop-opacity:1" />
+    </linearGradient>
+  </defs>
+
+  <!-- Background -->
+  <rect width="1280" height="640" fill="url(#bgGrad)"/>
+
+  <!-- Decorative background elements -->
+  <circle cx="100" cy="100" r="200" fill="#005CED" opacity="0.05"/>
+  <circle cx="1180" cy="540" r="250" fill="#00D4AA" opacity="0.05"/>
+
+  <!-- Logo group - smaller, positioned on left -->
+  <g transform="translate(120, 170)">
+    <!-- O - outer ring -->
+    <circle cx="150" cy="150" r="125" fill="none" stroke="url(#oGrad)" stroke-width="20"/>
+
+    <!-- Smooth S curve -->
+    <path d="M 172 80
+             Q 117 80, 112 117
+             Q 108 150, 150 150
+             Q 192 150, 188 188
+             Q 183 220, 126 220"
+          fill="none"
+          stroke="url(#sGrad)"
+          stroke-width="12"
+          stroke-linecap="round"/>
+
+    <!-- Nodes at key positions -->
+    <circle cx="172" cy="80" r="18" fill="#00D4AA"/>
+    <circle cx="112" cy="117" r="14" fill="#00CCBB"/>
+    <circle cx="150" cy="150" r="18" fill="#00C4C0"/>
+    <circle cx="188" cy="188" r="14" fill="#00BCC5"/>
+    <circle cx="126" cy="220" r="18" fill="#00B4D8"/>
+  </g>
+
+  <!-- Text "OnnxSlim" - larger, centered vertically -->
+  <text x="420" y="360" font-family="Arial, Helvetica, sans-serif" font-size="150" font-weight="bold" fill="url(#textGrad)">OnnxSlim</text>
+</svg>