m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

modify tiny-darknet

Browse Source
pull/10699/head
wukesong 4 years ago
parent 21d6722386
commit 4ca57dfa24
2 changed files with 28 additions and 14 deletions
Show Stats Download Patch File Download Diff File

21
model_zoo/official/cv/tinydarknet/README.md
Unescape View File

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

21
model_zoo/official/cv/tinydarknet/README_CN.md
Unescape View File

@ -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配置文件
```

Write Preview
Loading…
Cancel
Save