From 4ca57dfa244207b1f44005104adad4b3a02d1b75 Mon Sep 17 00:00:00 2001 From: wukesong Date: Sun, 27 Dec 2020 15:21:25 +0800 Subject: [PATCH] modify tiny-darknet --- model_zoo/official/cv/tinydarknet/README.md | 21 ++++++++++++------- .../official/cv/tinydarknet/README_CN.md | 21 ++++++++++++------- 2 files changed, 28 insertions(+), 14 deletions(-) diff --git a/model_zoo/official/cv/tinydarknet/README.md b/model_zoo/official/cv/tinydarknet/README.md index f173b0b8fa..fbac7f4354 100644 --- a/model_zoo/official/cv/tinydarknet/README.md +++ b/model_zoo/official/cv/tinydarknet/README.md @@ -96,20 +96,27 @@ For more details, please refer the specify script. ```bash -├── Tiny-DarkNet +├── tinydarknet ├── README.md // descriptions about Tiny-Darknet in English ├── README_CN.md // descriptions about Tiny-Darknet in Chinese ├── scripts - │ ├──run_standalone_train.sh // shell script for single on Ascend - │ ├──run_distribute_train.sh // shell script for distributed on Ascend - │ ├──run_eval.sh // shell script for evaluation on Ascend + ├──run_standalone_train.sh // shell script for single on Ascend + ├──run_distribute_train.sh // shell script for distributed on Ascend + ├──run_eval.sh // shell script for evaluation on Ascend ├── src - │ ├──dataset.py // creating dataset - │ ├──tinydarknet.py // Tiny-Darknet architecture - │ ├──config.py // parameter configuration + ├─lr_scheduler //learning rate scheduler + ├─__init__.py // init + ├─linear_warmup.py // linear_warmup + ├─warmup_cosine_annealing_lr.py // warmup_cosine_annealing_lr + ├─warmup_step_lr.py // warmup_step_lr + ├──dataset.py // creating dataset + ├──CrossEntropySmooth.py // loss function + ├──tinydarknet.py // Tiny-Darknet architecture + ├──config.py // parameter configuration ├── train.py // training script ├── eval.py // evaluation script ├── export.py // export checkpoint file into air/onnx + ├── mindspore_hub_conf.py // hub config ``` diff --git a/model_zoo/official/cv/tinydarknet/README_CN.md b/model_zoo/official/cv/tinydarknet/README_CN.md index 654c0a8220..890b9b4b87 100644 --- a/model_zoo/official/cv/tinydarknet/README_CN.md +++ b/model_zoo/official/cv/tinydarknet/README_CN.md @@ -104,20 +104,27 @@ Tiny-DarkNet是Joseph Chet Redmon等人提出的一个16层的针对于经典的 ```bash -├── Tiny-DarkNet +├── tinydarknet ├── README.md // Tiny-Darknet英文说明 ├── README_CN.md // Tiny-Darknet中文说明 ├── scripts - │ ├──run_standalone_train.sh // Ascend单卡训练shell脚本 - │ ├──run_distribute_train.sh // Ascend分布式训练shell脚本 - │ ├──run_eval.sh // Ascend评估shell脚本 + ├──run_standalone_train.sh // Ascend单卡训练shell脚本 + ├──run_distribute_train.sh // Ascend分布式训练shell脚本 + ├──run_eval.sh // Ascend评估shell脚本 ├── src - │ ├──dataset.py // 创建数据集 - │ ├──tinydarknet.py // Tiny-Darknet网络结构 - │ ├──config.py // 参数配置 + ├─lr_scheduler //学习率策略 + ├─__init__.py // 初始化文件 + ├─linear_warmup.py // linear_warmup策略 + ├─warmup_cosine_annealing_lr.py // warmup_cosine_annealing_lr策略 + ├─warmup_step_lr.py // warmup_step_lr策略 + ├──dataset.py // 创建数据集 + ├──CrossEntropySmooth.py // 损失函数 + ├──tinydarknet.py // Tiny-Darknet网络结构 + ├──config.py // 参数配置 ├── train.py // 训练脚本 ├── eval.py // 评估脚本 ├── export.py // 导出checkpoint文件 + ├── mindspore_hub_conf.py // hub配置文件 ```