WIP
This commit is contained in:
34
README.md
34
README.md
@@ -76,6 +76,38 @@
|
||||
| 8 | payment_line | 支付行 (机器可读格式) |
|
||||
| 9 | customer_number | 客户编号 |
|
||||
|
||||
## DPI 配置
|
||||
|
||||
**重要**: 系统所有组件统一使用 **150 DPI**,确保训练和推理的一致性。
|
||||
|
||||
DPI(每英寸点数)设置必须在训练和推理时保持一致,否则会导致:
|
||||
- 检测框尺寸失配
|
||||
- mAP显著下降(可能从93.5%降到60-70%)
|
||||
- 字段漏检或误检
|
||||
|
||||
### 配置位置
|
||||
|
||||
| 组件 | 配置文件 | 配置项 |
|
||||
|------|---------|--------|
|
||||
| **全局常量** | `src/config.py` | `DEFAULT_DPI = 150` |
|
||||
| **Web推理** | `src/web/config.py` | `ModelConfig.dpi` (导入自 `src.config`) |
|
||||
| **CLI推理** | `src/cli/infer.py` | `--dpi` 默认值 = `DEFAULT_DPI` |
|
||||
| **自动标注** | `src/config.py` | `AUTOLABEL['dpi'] = DEFAULT_DPI` |
|
||||
| **PDF转图** | `src/web/api/v1/admin/documents.py` | 使用 `DEFAULT_DPI` |
|
||||
|
||||
### 使用示例
|
||||
|
||||
```bash
|
||||
# 训练(使用默认150 DPI)
|
||||
python -m src.cli.autolabel --dual-pool --cpu-workers 3 --gpu-workers 1
|
||||
|
||||
# 推理(默认150 DPI,与训练一致)
|
||||
python -m src.cli.infer -m runs/train/invoice_fields/weights/best.pt -i invoice.pdf
|
||||
|
||||
# 手动指定DPI(仅当需要与非默认训练DPI的模型配合时)
|
||||
python -m src.cli.infer -m custom_model.pt -i invoice.pdf --dpi 150
|
||||
```
|
||||
|
||||
## 安装
|
||||
|
||||
```bash
|
||||
@@ -490,7 +522,7 @@ Options:
|
||||
--input, -i 输入 PDF/图像
|
||||
--output, -o 输出 JSON 路径
|
||||
--confidence 置信度阈值 (默认: 0.5)
|
||||
--dpi 渲染 DPI (默认: 300)
|
||||
--dpi 渲染 DPI (默认: 150, 必须与训练DPI一致)
|
||||
--gpu 使用 GPU
|
||||
```
|
||||
|
||||
|
||||
Reference in New Issue
Block a user