cjm-yolox-pytorch
Install
pip install cjm_yolox_pytorch
How to use
import torch
from cjm_yolox_pytorch.model import MODEL_TYPES, build_model
Select model type
model_type = MODEL_TYPES[0]
model_type
'yolox_tiny'
Build YOLOX model
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])]
1 maintainer
Metadata
1 maintainer
Release files for cjm-yolox-pytorch 0.2.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| cjm_yolox_pytorch-0.2.0.tar.gz | 23.7 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| cjm_yolox_pytorch-0.2.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 48.7 kB
Release files / cjm_yolox_pytorch-0.2.0.tar.gz
| Download URL | cjm_yolox_pytorch-0.2.0.tar.gz |
|---|---|
| Size | 23.7 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
f321c419b4cdd79233dae46b9f3f629b6af5068d224702e6fa1e35f8cd3f1b43
|
|
BLAKE2b-256 checksum How to use checksums |
2fb62d44a52f4bf30382cb4b634e9a2a611d2738863633c9567d07a56d390a3f
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.1 CPython/3.10.15
|
Release files / cjm_yolox_pytorch-0.2.0-py3-none-any.whl
| Download URL | cjm_yolox_pytorch-0.2.0-py3-none-any.whl |
|---|---|
| Size | 25.0 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
830b59d54cc25c26e339772f8198bd81bcaa7c72f5c970469dd2495ca307953b
|
|
BLAKE2b-256 checksum How to use checksums |
dbbd72281753520aefebab81d91dd2c047011b9a12da9afababef869c1841223
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/5.1.1 CPython/3.10.15
|
