提交 f095671e 编写于 作者: R RangeKing

Update kie_en.md

1. fix typo
2. add code block's type
上级 9f452b62
...@@ -17,21 +17,21 @@ This section provides a tutorial example on how to quickly use, train, and evalu ...@@ -17,21 +17,21 @@ This section provides a tutorial example on how to quickly use, train, and evalu
[Wildreceipt dataset](https://paperswithcode.com/dataset/wildreceipt) is used for this tutorial. It contains 1765 photos, with 25 classes, and 50000 text boxes, which can be downloaded by wget: [Wildreceipt dataset](https://paperswithcode.com/dataset/wildreceipt) is used for this tutorial. It contains 1765 photos, with 25 classes, and 50000 text boxes, which can be downloaded by wget:
``` ```shell
wget https://paddleocr.bj.bcebos.com/dygraph_v2.1/kie/wildreceipt.tar && tar xf wildreceipt.tar wget https://paddleocr.bj.bcebos.com/dygraph_v2.1/kie/wildreceipt.tar && tar xf wildreceipt.tar
``` ```
Download the pretrained model and predict the result: Download the pretrained model and predict the result:
``` ```shell
cd PaddleOCR/ cd PaddleOCR/
wget https://paddleocr.bj.bcebos.com/dygraph_v2.1/kie/kie_vgg16.tar && tar xf kie_vgg16.tar wget https://paddleocr.bj.bcebos.com/dygraph_v2.1/kie/kie_vgg16.tar && tar xf kie_vgg16.tar
python3.7 tools/infer_kie.py -c configs/kie/kie_unet_sdmgr.yml -o Global.checkpoints=kie_vgg16/best_accuracy Global.infer_img=../wildreceipt/1.txt python3.7 tools/infer_kie.py -c configs/kie/kie_unet_sdmgr.yml -o Global.checkpoints=kie_vgg16/best_accuracy Global.infer_img=../wildreceipt/1.txt
``` ```
The prediction result is saved as the folder`./output/sdmgr_kie/predicts_kie.txt`, and the visualization result is saved as the folder`/output/sdmgr_kie/kie_results/`. The prediction result is saved as `./output/sdmgr_kie/predicts_kie.txt`, and the visualization results are saved in the folder`/output/sdmgr_kie/kie_results/`.
The visualization result is shown in the figure below: The visualization results are shown in the figure below:
<div align="center"> <div align="center">
<img src="./imgs/0.png" width="800"> <img src="./imgs/0.png" width="800">
...@@ -41,14 +41,14 @@ The visualization result is shown in the figure below: ...@@ -41,14 +41,14 @@ The visualization result is shown in the figure below:
## 2. Model Training ## 2. Model Training
Create a softlink to the folder, `PaddleOCR/train_data`: Create a softlink to the folder, `PaddleOCR/train_data`:
``` ```shell
cd PaddleOCR/ && mkdir train_data && cd train_data cd PaddleOCR/ && mkdir train_data && cd train_data
ln -s ../../wildreceipt ./ ln -s ../../wildreceipt ./
``` ```
The configuration file used for training is `configs/kie/kie_unet_sdmgr.yml`. The default training data path in the configuration file is `train_data/wildreceipt`. After preparing the data, you can execute the model training with the following command: The configuration file used for training is `configs/kie/kie_unet_sdmgr.yml`. The default training data path in the configuration file is `train_data/wildreceipt`. After preparing the data, you can execute the model training with the following command:
``` ```shell
python3.7 tools/train.py -c configs/kie/kie_unet_sdmgr.yml -o Global.save_model_dir=./output/kie/ python3.7 tools/train.py -c configs/kie/kie_unet_sdmgr.yml -o Global.save_model_dir=./output/kie/
``` ```
<a name="3-----"></a> <a name="3-----"></a>
...@@ -57,7 +57,7 @@ python3.7 tools/train.py -c configs/kie/kie_unet_sdmgr.yml -o Global.save_model_ ...@@ -57,7 +57,7 @@ python3.7 tools/train.py -c configs/kie/kie_unet_sdmgr.yml -o Global.save_model_
After training, you can execute the model evaluation with the following command: After training, you can execute the model evaluation with the following command:
``` ```shell
python3.7 tools/eval.py -c configs/kie/kie_unet_sdmgr.yml -o Global.checkpoints=./output/kie/best_accuracy python3.7 tools/eval.py -c configs/kie/kie_unet_sdmgr.yml -o Global.checkpoints=./output/kie/best_accuracy
``` ```
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册