Commit c087d184 authored by sunxx1's avatar sunxx1
Browse files

修改运行脚本

parent fe34b3a7
# DBnet
# DBNet
## 模型介绍
......@@ -18,7 +18,7 @@ DBNet 模型网络结构主要分为 3 个模块:
在本测试中可以使用icdar2015数据集。
## DBnet训练
## DBNet训练
### 环境配置
......@@ -39,9 +39,9 @@ DBNet 模型网络结构主要分为 3 个模块:
测试数据使用的是icdar2015数据集,使用的加速卡是DCU Z100L。
| 卡数 | 精度 |
| 卡数 | 精度 |
| :--: | :-----------------------------------: |
| 1 | recall: 0.767070, precision: 0.894410 |
| 1 | recall: 0.767070, precision: 0.894410 |
### 历史版本
......@@ -50,4 +50,3 @@ https://developer.hpccube.com/codes/modelzoo/dbnet
### 参考
https://github.com/WenmuZhou/DBNet.pytorch
# 模型名称
modelName=DBnet
modelName=DBNet
# 模型描述
modelDescription=DBnet是一种使用可微分二值图来实时文字检测的深度学习模型
modelDescription=DBNet是一种使用可微分二值图来实时文字检测的深度学习模型
# 应用场景
appScenario=文本检测
# 框架类型
......
export HIP_VISIBLE_DEVICES=1
export HIP_VISIBLE_DEVICES=0
python3 tools/train.py --config_file "config/icdar2015_resnet18_FPN_DBhead_polyLR.yaml"
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment