Add model card
Browse files
README.md
CHANGED
|
@@ -1,3 +1,155 @@
|
|
| 1 |
-
---
|
| 2 |
-
license: mit
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
tags:
|
| 4 |
+
- object-detection
|
| 5 |
+
- yolox
|
| 6 |
+
- onnx
|
| 7 |
+
- int8
|
| 8 |
+
- coco
|
| 9 |
+
- fpga
|
| 10 |
+
language: en
|
| 11 |
+
library_name: onnxruntime
|
| 12 |
+
pipeline_tag: object-detection
|
| 13 |
+
---
|
| 14 |
+
|
| 15 |
+
# YOLOX-Nano ReLU (MIT) β work-in-progress training
|
| 16 |
+
|
| 17 |
+
A YOLOX-Nano variant with **ReLU activations and no depthwise convolutions**,
|
| 18 |
+
**trained from random initialisation on COCO 2017 by Pablo Mendoza
|
| 19 |
+
(`@thefalley`)** on his own server (NVIDIA GTX 1050 Ti, 4 GB VRAM). The
|
| 20 |
+
weights, the ONNX export pipeline, the INT8 quantization, the decoder and
|
| 21 |
+
all inference scripts in this repository are original work, released under
|
| 22 |
+
the MIT License.
|
| 23 |
+
|
| 24 |
+
The training architecture is `yolox_nano_ti_lite` from TexasInstruments'
|
| 25 |
+
`edgeai-yolox` repository β used as a build tool only, **not redistributed**
|
| 26 |
+
here. See `NOTICE.md` for the full provenance chain.
|
| 27 |
+
|
| 28 |
+
## Training status (live)
|
| 29 |
+
|
| 30 |
+
| | |
|
| 31 |
+
|---|---|
|
| 32 |
+
| Current epoch | **1 / 300** |
|
| 33 |
+
| AP @ IoU=0.5:0.95 | *** (will be measured at validation milestones) |
|
| 34 |
+
| Hardware | GTX 1050 Ti, 4 GB VRAM (own server, no cloud) |
|
| 35 |
+
| Started | 2026-05-10 |
|
| 36 |
+
| Last update | 2026-05-11 |
|
| 37 |
+
| Status | π Training in progress |
|
| 38 |
+
| Target | epoch 300, target mAP@0.5:0.95 β 0.26 (TI baseline for the same architecture) |
|
| 39 |
+
|
| 40 |
+
> **This is an early-checkpoint release** intended to validate the full
|
| 41 |
+
> pipeline (PTH β ONNX β INT8 β inference) end-to-end. Detection quality
|
| 42 |
+
> will improve substantially as training progresses; the repository will
|
| 43 |
+
> be updated incrementally with later checkpoints.
|
| 44 |
+
|
| 45 |
+
## Files
|
| 46 |
+
|
| 47 |
+
| File | Size | SHA-256 |
|
| 48 |
+
|---|---:|---|
|
| 49 |
+
| `yolox_nano_relu_float.onnx` | *** | *** |
|
| 50 |
+
| `yolox_nano_relu_int8_qop.onnx` | *** | *** |
|
| 51 |
+
|
| 52 |
+
## Architecture
|
| 53 |
+
|
| 54 |
+
| | |
|
| 55 |
+
|---|---|
|
| 56 |
+
| Family | YOLOX-Nano (Megvii, 2021) with the TI ti-lite modifications |
|
| 57 |
+
| Depth multiplier | 0.33 |
|
| 58 |
+
| Width multiplier | 0.25 |
|
| 59 |
+
| Parameters | ~1.9 M |
|
| 60 |
+
| Activation | **ReLU** (every Conv block; no SiLU) |
|
| 61 |
+
| Convolutions | **Regular Conv2D only** (no depthwise separable) |
|
| 62 |
+
| Input | 1Γ3Γ416Γ416, RGB, NCHW |
|
| 63 |
+
| Output (when exported with `--no-onnxsim`) | single tensor `(1, N, 85)` with `[cx, cy, w, h, obj, class_0..class_79]` already in input-pixel coords, anchor-free YOLOX-style |
|
| 64 |
+
| Quantization | Per-tensor INT8 (W symmetric, A asymmetric); bias INT32 |
|
| 65 |
+
|
| 66 |
+
These design choices are deliberate to match the operator set of an
|
| 67 |
+
INT8 FPGA DPU (Xilinx ZedBoard XC7Z020) that supports Conv2D + ReLU/Leaky
|
| 68 |
+
+ Add + Concat + MaxPool + Resize + Transpose only.
|
| 69 |
+
|
| 70 |
+
## Performance
|
| 71 |
+
|
| 72 |
+
| Metric | FP32 | INT8 | Reference (TI baseline) |
|
| 73 |
+
|---|---|---|---|
|
| 74 |
+
| AP @ IoU=0.5:0.95 | *** | *** | 0.261 |
|
| 75 |
+
| AP @ IoU=0.5 | *** | *** | 0.418 |
|
| 76 |
+
|
| 77 |
+
Reference numbers are TI's published metrics for the fully-trained
|
| 78 |
+
yolox_nano_ti_lite (300 epochs). Our weights are still being trained;
|
| 79 |
+
intermediate metrics will appear here as checkpoints are released.
|
| 80 |
+
|
| 81 |
+
## Visual inference samples
|
| 82 |
+
|
| 83 |
+
Sample detections produced by the current published checkpoint of this
|
| 84 |
+
model on classic test images (epoch 1 / 300 β detection quality will
|
| 85 |
+
improve as training advances).
|
| 86 |
+
|
| 87 |
+
| | |
|
| 88 |
+
|---|---|
|
| 89 |
+
|  |  |
|
| 90 |
+
|  |  |
|
| 91 |
+
|
| 92 |
+
## Reproducibility
|
| 93 |
+
|
| 94 |
+
This release is a snapshot of an ongoing training run. The pipeline:
|
| 95 |
+
|
| 96 |
+
```
|
| 97 |
+
1. Train (TI edgeai-yolox, GTX 1050 Ti, COCO train2017)
|
| 98 |
+
β yolox_nano_relu.pth
|
| 99 |
+
|
| 100 |
+
2. Export ONNX (tools/export_onnx.py from edgeai-yolox, --opset 13)
|
| 101 |
+
β yolox_nano_relu_float.onnx
|
| 102 |
+
|
| 103 |
+
3. Quantize INT8 (onnxruntime.quantize_static, 1000 random COCO val2017
|
| 104 |
+
images for calibration, seed = 42, per-tensor)
|
| 105 |
+
β yolox_nano_relu_int8_qop.onnx
|
| 106 |
+
```
|
| 107 |
+
|
| 108 |
+
Calibration was performed once on the float ONNX of the published
|
| 109 |
+
checkpoint and will be re-run for each later checkpoint to keep INT8
|
| 110 |
+
in sync with the trained float weights.
|
| 111 |
+
|
| 112 |
+
## Provenance summary
|
| 113 |
+
|
| 114 |
+
```
|
| 115 |
+
TI yolox_nano_ti_lite exp file (BSD-3 + Apache-2.0, build tool only)
|
| 116 |
+
β
|
| 117 |
+
β Train from scratch, COCO train2017, own hardware
|
| 118 |
+
βΌ
|
| 119 |
+
yolox_nano_relu.pth MIT (original training output)
|
| 120 |
+
β
|
| 121 |
+
β TI export_onnx.py (build tool)
|
| 122 |
+
βΌ
|
| 123 |
+
yolox_nano_relu_float.onnx MIT (this repository)
|
| 124 |
+
β
|
| 125 |
+
β onnxruntime.quantize_static (MIT, tool) + COCO val2017 (CC BY 4.0)
|
| 126 |
+
βΌ
|
| 127 |
+
yolox_nano_relu_int8_qop.onnx MIT (this repository)
|
| 128 |
+
```
|
| 129 |
+
|
| 130 |
+
The TI source code is used only as a build-time tool and is **not**
|
| 131 |
+
redistributed here. The weights themselves are an original training
|
| 132 |
+
output of the author. See `NOTICE.md` for full attribution.
|
| 133 |
+
|
| 134 |
+
## Detector size collection by the same author
|
| 135 |
+
|
| 136 |
+
| Repository | INT8 size | mAP@0.5:0.95 |
|
| 137 |
+
|---|---|---|
|
| 138 |
+
| [yolov4-leaky-416-int8-qop](https://huggingface.co/Thefalley/yolov4-leaky-416-int8-qop) | 61.66 MiB | 0.345 |
|
| 139 |
+
| [yolov4-tiny-416-int8-qop](https://huggingface.co/Thefalley/yolov4-tiny-416-int8-qop) | 5.83 MiB | 0.163 |
|
| 140 |
+
| [yolo-fastest-1.1-320-int8-qop](https://huggingface.co/Thefalley/yolo-fastest-1.1-320-int8-qop) | 0.47 MiB | (pending) |
|
| 141 |
+
| **yolox-nano-relu-mit** (this) | *** | *** (training in progress) |
|
| 142 |
+
|
| 143 |
+
## Citation
|
| 144 |
+
|
| 145 |
+
```bibtex
|
| 146 |
+
@article{ge2021yolox,
|
| 147 |
+
author = {Ge, Zheng and Liu, Songtao and Wang, Feng and Li, Zeming and Sun, Jian},
|
| 148 |
+
title = {YOLOX: Exceeding YOLO Series in 2021},
|
| 149 |
+
journal = {arXiv:2107.08430},
|
| 150 |
+
year = {2021}
|
| 151 |
+
}
|
| 152 |
+
```
|
| 153 |
+
|
| 154 |
+
Author of the trained weights and INT8 derivative: **Pablo Mendoza**
|
| 155 |
+
(`@thefalley`), 2026.
|