yolonnx


Nameyolonnx JSON
Version 0.1.5 PyPI version JSON
download
home_pagehttps://www.cs.aau.dk/
SummaryThis package lets you run your YOLOv8 detection and classification models using ONNXRuntime.
upload_time2024-05-21 10:23:43
maintainerNone
docs_urlNone
authorNone
requires_python>=3.10
licenseMIT
keywords onnx yolov8 onnxruntime vision
VCS
bugtrack_url
requirements No requirements were recorded.
Travis-CI No Travis.
coveralls test coverage No coveralls.
            # You Only Look ONNX
This repository is a light weight library to ease the use of ONNX models exported by the Ultralytics YOLOv8 framework.

## Example Detector Usage
```python
from pathlib import Path

from onnxruntime import InferenceSession
from PIL import Image

from yolonnx.services import Detector
from yolonnx.to_tensor_strategies import PillowToTensorContainStrategy

model = Path("path/to/file.onnx")
session = InferenceSession(
    model.as_posix(),
    providers=[
        "CUDAExecutionProvider",
        "CPUExecutionProvider",
    ],
)

predictor = Detector(session, PillowToTensorContainStrategy())
img = Image.open("path/to/image.jpg")
print(predictor.run(img))
```

## Example Classifier Usage
```python
from pathlib import Path

from onnxruntime import InferenceSession
from PIL import Image

from yolonnx.services import Classifier
from yolonnx.to_tensor_strategies import PillowToTensorContainStrategy

model = Path("path/to/file.onnx")
session = InferenceSession(
    model.as_posix(),
    providers=[
        "CUDAExecutionProvider",
        "CPUExecutionProvider",
    ],
)

predictor = Classifier(session, PillowToTensorContainStrategy())
img = Image.open("path/to/image.jpg")
print(predictor.run(img))

```
            

Raw data

            {
    "_id": null,
    "home_page": "https://www.cs.aau.dk/",
    "name": "yolonnx",
    "maintainer": null,
    "docs_url": null,
    "requires_python": ">=3.10",
    "maintainer_email": null,
    "keywords": "ONNX, YOLOv8, onnxruntime, vision",
    "author": null,
    "author_email": "Kasper Fromm Pedersen <kasperf@cs.aau.dk>",
    "download_url": "https://files.pythonhosted.org/packages/e7/81/641e63e8676e51fe61dbe52334d95ce6eeb365fb04c96f5ebc8ca36b8b76/yolonnx-0.1.5.tar.gz",
    "platform": null,
    "description": "# You Only Look ONNX\nThis repository is a light weight library to ease the use of ONNX models exported by the Ultralytics YOLOv8 framework.\n\n## Example Detector Usage\n```python\nfrom pathlib import Path\n\nfrom onnxruntime import InferenceSession\nfrom PIL import Image\n\nfrom yolonnx.services import Detector\nfrom yolonnx.to_tensor_strategies import PillowToTensorContainStrategy\n\nmodel = Path(\"path/to/file.onnx\")\nsession = InferenceSession(\n    model.as_posix(),\n    providers=[\n        \"CUDAExecutionProvider\",\n        \"CPUExecutionProvider\",\n    ],\n)\n\npredictor = Detector(session, PillowToTensorContainStrategy())\nimg = Image.open(\"path/to/image.jpg\")\nprint(predictor.run(img))\n```\n\n## Example Classifier Usage\n```python\nfrom pathlib import Path\n\nfrom onnxruntime import InferenceSession\nfrom PIL import Image\n\nfrom yolonnx.services import Classifier\nfrom yolonnx.to_tensor_strategies import PillowToTensorContainStrategy\n\nmodel = Path(\"path/to/file.onnx\")\nsession = InferenceSession(\n    model.as_posix(),\n    providers=[\n        \"CUDAExecutionProvider\",\n        \"CPUExecutionProvider\",\n    ],\n)\n\npredictor = Classifier(session, PillowToTensorContainStrategy())\nimg = Image.open(\"path/to/image.jpg\")\nprint(predictor.run(img))\n\n```",
    "bugtrack_url": null,
    "license": "MIT",
    "summary": "This package lets you run your YOLOv8 detection and classification models using ONNXRuntime.",
    "version": "0.1.5",
    "project_urls": {
        "Homepage": "https://www.cs.aau.dk/",
        "Repository": "https://github.com/fromm1990/yolonnx"
    },
    "split_keywords": [
        "onnx",
        " yolov8",
        " onnxruntime",
        " vision"
    ],
    "urls": [
        {
            "comment_text": "",
            "digests": {
                "blake2b_256": "de636b4bb6acc3fcdc9092e614d759e3959421c102abfe130d1688b0cbb00c34",
                "md5": "585febbc06da507f60520aba692c5677",
                "sha256": "9cd8bed177c105843b4b837430fb723395e96b824f1c80d63dd22a955c075ce6"
            },
            "downloads": -1,
            "filename": "yolonnx-0.1.5-py3-none-any.whl",
            "has_sig": false,
            "md5_digest": "585febbc06da507f60520aba692c5677",
            "packagetype": "bdist_wheel",
            "python_version": "py3",
            "requires_python": ">=3.10",
            "size": 7805,
            "upload_time": "2024-05-21T10:23:42",
            "upload_time_iso_8601": "2024-05-21T10:23:42.038940Z",
            "url": "https://files.pythonhosted.org/packages/de/63/6b4bb6acc3fcdc9092e614d759e3959421c102abfe130d1688b0cbb00c34/yolonnx-0.1.5-py3-none-any.whl",
            "yanked": false,
            "yanked_reason": null
        },
        {
            "comment_text": "",
            "digests": {
                "blake2b_256": "e781641e63e8676e51fe61dbe52334d95ce6eeb365fb04c96f5ebc8ca36b8b76",
                "md5": "ae72764a7440b350495ccb6a30943a5c",
                "sha256": "dc5adb7569681256779482861718504de30326e1958dc61443dfe7c848862db6"
            },
            "downloads": -1,
            "filename": "yolonnx-0.1.5.tar.gz",
            "has_sig": false,
            "md5_digest": "ae72764a7440b350495ccb6a30943a5c",
            "packagetype": "sdist",
            "python_version": "source",
            "requires_python": ">=3.10",
            "size": 5780,
            "upload_time": "2024-05-21T10:23:43",
            "upload_time_iso_8601": "2024-05-21T10:23:43.475257Z",
            "url": "https://files.pythonhosted.org/packages/e7/81/641e63e8676e51fe61dbe52334d95ce6eeb365fb04c96f5ebc8ca36b8b76/yolonnx-0.1.5.tar.gz",
            "yanked": false,
            "yanked_reason": null
        }
    ],
    "upload_time": "2024-05-21 10:23:43",
    "github": true,
    "gitlab": false,
    "bitbucket": false,
    "codeberg": false,
    "github_user": "fromm1990",
    "github_project": "yolonnx",
    "travis_ci": false,
    "coveralls": false,
    "github_actions": false,
    "lcname": "yolonnx"
}
        
Elapsed time: 0.25269s