From d135cc98d1bd93b18eb15859f4a6fdbf8a038cf8 Mon Sep 17 00:00:00 2001 From: zhouyaqiang Date: Fri, 15 Jan 2021 11:03:25 +0800 Subject: [PATCH] Modify readme of deeptext --- model_zoo/official/cv/deeptext/README.md | 19 +++++++++---------- .../cv/deeptext/scripts/run_eval_ascend.sh | 6 +++--- .../scripts/run_standalone_train_ascend.sh | 6 +++--- 3 files changed, 15 insertions(+), 16 deletions(-) diff --git a/model_zoo/official/cv/deeptext/README.md b/model_zoo/official/cv/deeptext/README.md index 29496af748..830d018e7a 100644 --- a/model_zoo/official/cv/deeptext/README.md +++ b/model_zoo/official/cv/deeptext/README.md @@ -41,7 +41,7 @@ Here we used 4 datasets for training, and 1 datasets for Evaluation. - Train: 27.7MB, 410 images - Dataset3: SCUT-FORU: Flickr OCR Universal Database - Train: 388MB, 1715 images -- Dataset4: CocoText v2(Subset of MSCOCO2014): +- Dataset4: CocoText v2(Subset of MSCOCO2017): - Train: 13GB, 63686 images # [Features](#contents) @@ -100,9 +100,9 @@ Here we used 4 datasets for training, and 1 datasets for Evaluation. # distribute training example(8p) sh run_distribute_train_ascend.sh [IMGS_PATH] [ANNOS_PATH] [RANK_TABLE_FILE] [PRETRAINED_PATH] [COCO_TEXT_PARSER_PATH] # standalone training -sh run_standalone_train_ascend.sh [IMGS_PATH] [ANNOS_PATH] [PRETRAINED_PATH] [COCO_TEXT_PARSER_PATH] +sh run_standalone_train_ascend.sh [IMGS_PATH] [ANNOS_PATH] [PRETRAINED_PATH] [COCO_TEXT_PARSER_PATH] [DEVICE_ID] # evaluation: -sh run_eval_ascend.sh [IMGS_PATH] [ANNOS_PATH] [CHECKPOINT_PATH] [COCO_TEXT_PARSER_PATH] +sh run_eval_ascend.sh [IMGS_PATH] [ANNOS_PATH] [CHECKPOINT_PATH] [COCO_TEXT_PARSER_PATH] [DEVICE_ID] ``` > Notes: @@ -122,7 +122,7 @@ sh run_eval_ascend.sh [IMGS_PATH] [ANNOS_PATH] [CHECKPOINT_PATH] [COCO_TEXT_PARS # distribute training example(8p) sh run_distribute_train_ascend.sh [IMGS_PATH] [ANNOS_PATH] [RANK_TABLE_FILE] [PRETRAINED_PATH] [COCO_TEXT_PARSER_PATH] # standalone training - sh run_standalone_train_ascend.sh [IMGS_PATH] [ANNOS_PATH] [PRETRAINED_PATH] [COCO_TEXT_PARSER_PATH] + sh run_standalone_train_ascend.sh [IMGS_PATH] [ANNOS_PATH] [PRETRAINED_PATH] [COCO_TEXT_PARSER_PATH] [DEVICE_ID] ``` ### Result @@ -144,7 +144,7 @@ You can start training using python or shell scripts. The usage of shell scripts - Ascend: ```bash - sh run_eval_ascend.sh [IMGS_PATH] [ANNOS_PATH] [CHECKPOINT_PATH] [COCO_TEXT_PARSER_PATH] + sh run_eval_ascend.sh [IMGS_PATH] [ANNOS_PATH] [CHECKPOINT_PATH] [COCO_TEXT_PARSER_PATH] [DEVICE_ID] ``` ### Launch @@ -153,7 +153,7 @@ You can start training using python or shell scripts. The usage of shell scripts # eval example shell: Ascend: - sh run_eval_ascend.sh [IMGS_PATH] [ANNOS_PATH] [CHECKPOINT_PATH] [COCO_TEXT_PARSER_PATH] + sh run_eval_ascend.sh [IMGS_PATH] [ANNOS_PATH] [CHECKPOINT_PATH] [COCO_TEXT_PARSER_PATH] [DEVICE_ID] ``` > checkpoint can be produced in training process. @@ -186,7 +186,6 @@ class 1 precision is 88.01%, recall is 82.77% | Optimizer | Momentum | | Loss Function | SoftmaxCrossEntropyWithLogits for classification, SmoothL2Loss for bbox regression| | Loss | ~0.008 | -| Accuracy (8p) | precision=0.8854, recall=0.8024 | | Total time (8p) | 4h | | Scripts | [deeptext script](https://gitee.com/mindspore/mindspore/tree/r1.1/mindspore/official/cv/deeptext) | @@ -200,7 +199,7 @@ class 1 precision is 88.01%, recall is 82.77% | MindSpore Version | 1.1.0 | | Dataset | 229 images | | Batch_size | 2 | -| Accuracy | precision=0.8854, recall=0.8024 | +| Accuracy | precision=0.8801, recall=0.8277 | | Total time | 1 min | | Model for inference | 3492M (.ckpt file) | @@ -208,11 +207,11 @@ class 1 precision is 88.01%, recall is 82.77% | **Ascend** | train performance | | :--------: | :---------------: | -| 1p | 42 img/s | +| 1p | 14 img/s | | **Ascend** | train performance | | :--------: | :---------------: | -| 8p | 330 img/s | +| 8p | 50 img/s | # [Description of Random Situation](#contents) diff --git a/model_zoo/official/cv/deeptext/scripts/run_eval_ascend.sh b/model_zoo/official/cv/deeptext/scripts/run_eval_ascend.sh index 7be97c6a8a..979d1d0719 100644 --- a/model_zoo/official/cv/deeptext/scripts/run_eval_ascend.sh +++ b/model_zoo/official/cv/deeptext/scripts/run_eval_ascend.sh @@ -14,9 +14,9 @@ # limitations under the License. # ============================================================================ -if [ $# != 4 ] +if [ $# != 5 ] then - echo "Usage: sh run_eval_ascend.sh [IMGS_PATH] [ANNOS_PATH] [CHECKPOINT_PATH] [COCO_TEXT_PARSER_PATH]" + echo "Usage: sh run_eval_ascend.sh [IMGS_PATH] [ANNOS_PATH] [CHECKPOINT_PATH] [COCO_TEXT_PARSER_PATH] [DEVICE_ID]" exit 1 fi @@ -52,7 +52,7 @@ fi ulimit -u unlimited export DEVICE_NUM=1 export RANK_SIZE=$DEVICE_NUM -export DEVICE_ID=0 +export DEVICE_ID=$5 export RANK_ID=0 cp $PATH4 ../src/ if [ -d "eval" ]; diff --git a/model_zoo/official/cv/deeptext/scripts/run_standalone_train_ascend.sh b/model_zoo/official/cv/deeptext/scripts/run_standalone_train_ascend.sh index 114239f17d..a6f2f24238 100644 --- a/model_zoo/official/cv/deeptext/scripts/run_standalone_train_ascend.sh +++ b/model_zoo/official/cv/deeptext/scripts/run_standalone_train_ascend.sh @@ -14,9 +14,9 @@ # limitations under the License. # ============================================================================ -if [ $# -ne 4 ] +if [ $# -ne 5 ] then - echo "Usage: sh run_standalone_train_ascend.sh [IMGS_PATH] [ANNOS_PATH] [PRETRAINED_PATH] [COCO_TEXT_PARSER_PATH]" + echo "Usage: sh run_standalone_train_ascend.sh [IMGS_PATH] [ANNOS_PATH] [PRETRAINED_PATH] [COCO_TEXT_PARSER_PATH] [DEVICE_ID]" exit 1 fi @@ -51,7 +51,7 @@ fi ulimit -u unlimited export DEVICE_NUM=1 -export DEVICE_ID=0 +export DEVICE_ID=$5 export RANK_ID=0 export RANK_SIZE=1 cp $PATH4 ../src/