diff --git a/model_zoo/official/cv/psenet/README.md b/model_zoo/official/cv/psenet/README.md
index 6874f80ba4..8893fbab64 100644
--- a/model_zoo/official/cv/psenet/README.md
+++ b/model_zoo/official/cv/psenet/README.md
@@ -105,7 +105,10 @@ sh scripts/run_eval_ascend.sh
                 ├── Makefile
 		├── config.py                       // parameter configuration 
 		├── dataset.py                      // creating dataset
+		├── lr_schedule.py                  // learning ratio generation
 		└── network_define.py               // PSENet architecture
+	├── export.py                           // export mindir file 
+	├── mindspore_hub_conf.py               // hub config file 
 	├── test.py                             // test script 
 	└── train.py                            // training script
 
@@ -132,6 +135,7 @@ Major parameters in train.py and config.py are:
 sh scripts/run_distribute_train.sh rank_table_file pretrained_model.ckpt
 ```
 
+rank_table_file which is specified by RANK_TABLE_FILE is needed when you are running a distribute task. You can generate it by using the [hccl_tools](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/utils/hccl_tools).
 The above shell script will run distribute training in the background. You can view the results through the file 
 `device[X]/test_*.log`. The loss value will be achieved as follows:
 
diff --git a/model_zoo/official/cv/psenet/scripts/__init__.py b/model_zoo/official/cv/psenet/scripts/__init__.py
deleted file mode 100644
index e30774307c..0000000000
--- a/model_zoo/official/cv/psenet/scripts/__init__.py
+++ /dev/null
@@ -1,14 +0,0 @@
-# Copyright 2020 Huawei Technologies Co., Ltd
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ============================================================================
diff --git a/model_zoo/official/cv/psenet/src/ETSNET/pse/adaptor.cpp b/model_zoo/official/cv/psenet/src/ETSNET/pse/adaptor.cpp
index a7092b4c5d..0922d9f8cf 100644
--- a/model_zoo/official/cv/psenet/src/ETSNET/pse/adaptor.cpp
+++ b/model_zoo/official/cv/psenet/src/ETSNET/pse/adaptor.cpp
@@ -28,6 +28,7 @@
 
 using std::vector;
 using std::queue;
+using std::swap;
 using cv::Mat;
 using cv::Point;