Skip to content

Commit 6144a50

Browse files
authored
partial refactoring of py_infer & OCRServer (#638)
* Fix: result of det infer node * Fix det infer
1 parent c700d02 commit 6144a50

28 files changed

+807
-226
lines changed

configs/det/fcenet/README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -104,17 +104,17 @@ train:
104104
dataset:
105105
type: DetDataset
106106
dataset_root: dir/to/dataset <--- Update
107-
data_dir: icda/ch4_training_images <--- Update
108-
label_file: icda/train_det_gt.txt <--- Update
107+
data_dir: train/images <--- Update
108+
label_file: train/train_det_gt.txt <--- Update
109109
...
110110
eval:
111111
ckpt_load_path: '/best.ckpt' <--- Update
112112
dataset_sink_mode: False
113113
dataset:
114114
type: DetDataset
115115
dataset_root: dir/to/dataset <--- Update
116-
data_dir: icda/ch4_test_images <--- Update
117-
label_file: icda/test_det_gt.txt <--- Update
116+
data_dir: test/images <--- Update
117+
label_file: test/test_det_gt.txt <--- Update
118118
...
119119
```
120120

configs/det/fcenet/README_CN.md

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -103,23 +103,23 @@ MindOCR中的FCENet网络在ICDAR 2015数据集上训练。训练结果如下:
103103
```yaml
104104
...
105105
train:
106-
ckpt_save_dir: './tmp_det_fcenet'
106+
ckpt_save_dir: './tmp_det'
107107
dataset_sink_mode: False
108108
ema: True
109109
dataset:
110110
type: DetDataset
111111
dataset_root: dir/to/dataset <--- 更新
112-
data_dir: icda/ch4_training_images <--- 更新
113-
label_file: icda/train_det_gt.txt <--- 更新
112+
data_dir: train/images <--- 更新
113+
label_file: train/train_det_gt.txt <--- 更新
114114
...
115115
eval:
116116
ckpt_load_path: '/best.ckpt' <--- 更新
117117
dataset_sink_mode: False
118118
dataset:
119119
type: DetDataset
120120
dataset_root: dir/to/dataset <--- 更新
121-
data_dir: icda/ch4_test_images <--- 更新
122-
label_file: icda/test_det_gt.txt <--- 更新
121+
data_dir: test/images <--- 更新
122+
label_file: test/test_det_gt.txt <--- 更新
123123
...
124124
```
125125

configs/det/fcenet/fce_icdar15.yaml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -52,14 +52,14 @@ optimizer:
5252

5353

5454
train:
55-
ckpt_save_dir: './tmp_det_fcenet'
55+
ckpt_save_dir: './tmp_det'
5656
dataset_sink_mode: True
5757
ema: True
5858
dataset:
5959
type: DetDataset
60-
dataset_root: data
61-
data_dir: icda/ch4_training_images
62-
label_file: icda/train_det_gt.txt
60+
dataset_root: /data/ocr_datasets
61+
data_dir: ic15/det/train/ch4_training_images
62+
label_file: ic15/det/train/det_gt.txt
6363
sample_ratio: 1.0
6464
transform_pipeline:
6565
- DecodeImage:
@@ -114,9 +114,9 @@ eval:
114114
dataset_sink_mode: False
115115
dataset:
116116
type: DetDataset
117-
dataset_root: data
118-
data_dir: icda/ch4_test_images
119-
label_file: icda/test_det_gt.txt
117+
dataset_root: /data/ocr_datasets
118+
data_dir: ic15/det/test/ch4_test_images
119+
label_file: ic15/det/test/det_gt.txt
120120
sample_ratio: 1.0
121121
shuffle: False
122122
transform_pipeline:
5.86 KB
Loading
4.32 KB
Loading
11.2 KB
Loading
6.34 KB
Loading
12.4 KB
Loading

0 commit comments

Comments
 (0)