# cjm-yolox-pytorch
<!-- WARNING: THIS FILE WAS AUTOGENERATED! DO NOT EDIT! -->
## Install
``` sh
pip install cjm_yolox_pytorch
```
## How to use
``` python
import torch
from cjm_yolox_pytorch.model import MODEL_TYPES, build_model
```
**Select model type**
``` python
model_type = MODEL_TYPES[0]
model_type
```
'yolox_tiny'
**Build YOLOX model**
``` python
yolox = build_model(model_type, 19, pretrained=True)
test_inp = torch.randn(1, 3, 256, 256)
with torch.no_grad():
cls_scores, bbox_preds, objectness = yolox(test_inp)
print(f"cls_scores: {[cls_score.shape for cls_score in cls_scores]}")
print(f"bbox_preds: {[bbox_pred.shape for bbox_pred in bbox_preds]}")
print(f"objectness: {[objectness.shape for objectness in objectness]}")
```
The file ./pretrained_checkpoints/yolox_tiny.pth already exists and overwrite is set to False.
cls_scores: [torch.Size([1, 19, 32, 32]), torch.Size([1, 19, 16, 16]), torch.Size([1, 19, 8, 8])]
bbox_preds: [torch.Size([1, 4, 32, 32]), torch.Size([1, 4, 16, 16]), torch.Size([1, 4, 8, 8])]
objectness: [torch.Size([1, 1, 32, 32]), torch.Size([1, 1, 16, 16]), torch.Size([1, 1, 8, 8])]
Raw data
{
"_id": null,
"home_page": "https://github.com/cj-mills/cjm-yolox-pytorch",
"name": "cjm-yolox-pytorch",
"maintainer": null,
"docs_url": null,
"requires_python": ">=3.9",
"maintainer_email": null,
"keywords": "nbdev jupyter notebook python torch",
"author": "cj-mills",
"author_email": "millscj.mills2@gmail.com",
"download_url": "https://files.pythonhosted.org/packages/2f/b6/2d44a52f4bf30382cb4b634e9a2a611d2738863633c9567d07a56d390a3f/cjm_yolox_pytorch-0.2.0.tar.gz",
"platform": null,
"description": "# cjm-yolox-pytorch\n\n\n<!-- WARNING: THIS FILE WAS AUTOGENERATED! DO NOT EDIT! -->\n\n## Install\n\n``` sh\npip install cjm_yolox_pytorch\n```\n\n## How to use\n\n``` python\nimport torch\nfrom cjm_yolox_pytorch.model import MODEL_TYPES, build_model\n```\n\n**Select model type**\n\n``` python\nmodel_type = MODEL_TYPES[0]\nmodel_type\n```\n\n 'yolox_tiny'\n\n**Build YOLOX model**\n\n``` python\nyolox = build_model(model_type, 19, pretrained=True)\n\ntest_inp = torch.randn(1, 3, 256, 256)\n\nwith torch.no_grad():\n cls_scores, bbox_preds, objectness = yolox(test_inp)\n \nprint(f\"cls_scores: {[cls_score.shape for cls_score in cls_scores]}\")\nprint(f\"bbox_preds: {[bbox_pred.shape for bbox_pred in bbox_preds]}\")\nprint(f\"objectness: {[objectness.shape for objectness in objectness]}\")\n```\n\n The file ./pretrained_checkpoints/yolox_tiny.pth already exists and overwrite is set to False.\n cls_scores: [torch.Size([1, 19, 32, 32]), torch.Size([1, 19, 16, 16]), torch.Size([1, 19, 8, 8])]\n bbox_preds: [torch.Size([1, 4, 32, 32]), torch.Size([1, 4, 16, 16]), torch.Size([1, 4, 8, 8])]\n objectness: [torch.Size([1, 1, 32, 32]), torch.Size([1, 1, 16, 16]), torch.Size([1, 1, 8, 8])]\n",
"bugtrack_url": null,
"license": "Apache Software License 2.0",
"summary": "A PyTorch implementation of the YOLOX object detection model based on the mmdetection implementation.",
"version": "0.2.0",
"project_urls": {
"Homepage": "https://github.com/cj-mills/cjm-yolox-pytorch"
},
"split_keywords": [
"nbdev",
"jupyter",
"notebook",
"python",
"torch"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "dbbd72281753520aefebab81d91dd2c047011b9a12da9afababef869c1841223",
"md5": "076347c0ecc6a1febc4196284c574d66",
"sha256": "830b59d54cc25c26e339772f8198bd81bcaa7c72f5c970469dd2495ca307953b"
},
"downloads": -1,
"filename": "cjm_yolox_pytorch-0.2.0-py3-none-any.whl",
"has_sig": false,
"md5_digest": "076347c0ecc6a1febc4196284c574d66",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": ">=3.9",
"size": 24992,
"upload_time": "2024-10-06T18:35:18",
"upload_time_iso_8601": "2024-10-06T18:35:18.847574Z",
"url": "https://files.pythonhosted.org/packages/db/bd/72281753520aefebab81d91dd2c047011b9a12da9afababef869c1841223/cjm_yolox_pytorch-0.2.0-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "2fb62d44a52f4bf30382cb4b634e9a2a611d2738863633c9567d07a56d390a3f",
"md5": "22846332c02cd6d5c1c00d7cff1b66c3",
"sha256": "f321c419b4cdd79233dae46b9f3f629b6af5068d224702e6fa1e35f8cd3f1b43"
},
"downloads": -1,
"filename": "cjm_yolox_pytorch-0.2.0.tar.gz",
"has_sig": false,
"md5_digest": "22846332c02cd6d5c1c00d7cff1b66c3",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.9",
"size": 23672,
"upload_time": "2024-10-06T18:35:20",
"upload_time_iso_8601": "2024-10-06T18:35:20.736353Z",
"url": "https://files.pythonhosted.org/packages/2f/b6/2d44a52f4bf30382cb4b634e9a2a611d2738863633c9567d07a56d390a3f/cjm_yolox_pytorch-0.2.0.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-10-06 18:35:20",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "cj-mills",
"github_project": "cjm-yolox-pytorch",
"travis_ci": false,
"coveralls": false,
"github_actions": true,
"lcname": "cjm-yolox-pytorch"
}
JSON
