Skip to content

GitLab

Commit c087d184 authored by sunxx1's avatar sunxx1
Browse files

修改运行脚本

parent fe34b3a7
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 8 deletions
README.md
View file @ c087d184
# DBnet # DBNet
## 模型介绍 ## 模型介绍
...@@ -18,7 +18,7 @@ DBNet 模型网络结构主要分为 3 个模块： ...@@ -18,7 +18,7 @@ DBNet 模型网络结构主要分为 3 个模块：
在本测试中可以使用icdar2015数据集。 在本测试中可以使用icdar2015数据集。
## DBnet训练 ## DBNet训练
### 环境配置 ### 环境配置
...@@ -39,9 +39,9 @@ DBNet 模型网络结构主要分为 3 个模块： ...@@ -39,9 +39,9 @@ DBNet 模型网络结构主要分为 3 个模块：
测试数据使用的是icdar2015数据集，使用的加速卡是DCU Z100L。 测试数据使用的是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 ...@@ -50,4 +50,3 @@ https://developer.hpccube.com/codes/modelzoo/dbnet
### 参考 ### 参考
https://github.com/WenmuZhou/DBNet.pytorch https://github.com/WenmuZhou/DBNet.pytorch
model.properties
View file @ c087d184
# 模型名称 # 模型名称
modelName=DBnet modelName=DBNet
# 模型描述 # 模型描述
modelDescription=DBnet是一种使用可微分二值图来实时文字检测的深度学习模型 modelDescription=DBNet是一种使用可微分二值图来实时文字检测的深度学习模型
# 应用场景 # 应用场景
appScenario=文本检测 appScenario=文本检测
# 框架类型 # 框架类型
......
run.sh
View file @ c087d184
export HIP_VISIBLE_DEVICES=1 export HIP_VISIBLE_DEVICES=0
python3 tools/train.py --config_file "config/icdar2015_resnet18_FPN_DBhead_polyLR.yaml" 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