docs: update model card (cleaned-label DET / seed1 TIA-on REC, 2026-06-02 release)
Browse files
README.md
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
language:
|
| 4 |
+
- ja
|
| 5 |
+
tags:
|
| 6 |
+
- ocr
|
| 7 |
+
- paddleocr
|
| 8 |
+
- steel-code
|
| 9 |
+
- recognition
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# Steel OCR REC Model
|
| 13 |
+
|
| 14 |
+
鉄骨製品コードを認識する en_PP-OCRv5_mobile_rec ファインチューニング済みモデル
|
| 15 |
+
(カスタム 21 文字辞書 `-0123456789ABGMNSXYbs`、CTC + GTC/NRTR aux head)。
|
| 16 |
+
パイプライン: 画像 → DET → crop → **REC(本モデル)** → 候補マッチング。
|
| 17 |
+
|
| 18 |
+
## 最新版: `tia-on-seed1-20260602`(現 main の `rec_model.onnx`)
|
| 19 |
+
|
| 20 |
+
de-leaked split(train222 / val95)で再学習した variance recipe + **TIA on** モデル(seed1)。
|
| 21 |
+
TIA(RecAug mesh warp)は recall に **+4.9pt(n=3 paired)** 寄与すると確認済。
|
| 22 |
+
|
| 23 |
+
| 指標 | 値 |
|
| 24 |
+
|---|---|
|
| 25 |
+
| clean-val raw_code_recall | 55.94(80/143) |
|
| 26 |
+
| clean-val matched_code_recall | 56.64 |
|
| 27 |
+
| E2E test(クリーン DET 併用)matched_recall | 61.54(旧 53.85、**+7.7pt**) |
|
| 28 |
+
|
| 29 |
+
### レシピ
|
| 30 |
+
reconaug on / jitter on / lr 5e-4 / warmup 2 / epoch 80 / batch 128 / **tia_prob 0.4** / Global.seed 1。
|
| 31 |
+
|
| 32 |
+
## ⚠ 重要(形式の注意)
|
| 33 |
+
- **`rec_model.onnx` が seed1 TIA-on 再学習版**(ONNX パイプラインはこれを使用)。
|
| 34 |
+
- **`best_accuracy.pdparams` / `inference/` は従来 stable のまま**(seed1 では未更新)。pdparams から
|
| 35 |
+
再 export すると seed1 でなく旧 stable になる点に注意。**辞書(`inference.yml` の 21 文字)は同一**。
|
| 36 |
+
|
| 37 |
+
## ファイル
|
| 38 |
+
- `rec_model.onnx` — ONNX(seed1 TIA-on・推奨)
|
| 39 |
+
- `inference/inference.yml` — 文字辞書(21 文字、ONNX パイプラインが自動読込)
|
| 40 |
+
|
| 41 |
+
## Usage
|
| 42 |
+
```python
|
| 43 |
+
from huggingface_hub import hf_hub_download
|
| 44 |
+
onnx = hf_hub_download("iput-tk230215/steel-ocr-rec", "rec_model.onnx",
|
| 45 |
+
revision="tia-on-seed1-20260602")
|
| 46 |
+
```
|
| 47 |
+
|
| 48 |
+
## Versions
|
| 49 |
+
[Tags](https://huggingface.co/iput-tk230215/steel-ocr-rec/tags) 参照。旧版は `stable-20260118`。
|
| 50 |
+
|
| 51 |
+
## Repository
|
| 52 |
+
https://github.com/iput-tk230215/solution_teamB_image-recog
|