psenet readme update

This commit is contained in:
linqingke 2020-11-09 09:46:34 +08:00
parent 370e7ab95f
commit 98a29c5308
3 changed files with 5 additions and 14 deletions

View File

@ -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:

View File

@ -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.
# ============================================================================

View File

@ -28,6 +28,7 @@
using std::vector;
using std::queue;
using std::swap;
using cv::Mat;
using cv::Point;