Thefalley commited on
Commit
2236bcc
Β·
verified Β·
1 Parent(s): ed69efb

Add model card

Browse files
Files changed (1) hide show
  1. README.md +155 -3
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
+ | ![dog](images/float_dog.png) | ![traffic](images/float_traffic.png) |
90
+ | ![parking](images/float_parking.png) | ![dining](images/float_dining.png) |
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.