@@ -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:
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:
<divalign="center">
<divalign="center">
<imgsrc="./imgs/0.png"width="800">
<imgsrc="./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: