Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleOCR
提交
69df224c
P
PaddleOCR
项目概览
PaddlePaddle
/
PaddleOCR
大约 1 年 前同步成功
通知
1528
Star
32962
Fork
6643
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
108
列表
看板
标记
里程碑
合并请求
7
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleOCR
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
108
Issue
108
列表
看板
标记
里程碑
合并请求
7
合并请求
7
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
69df224c
编写于
7月 21, 2022
作者:
qq_25193841
浏览文件
操作
浏览文件
下载
差异文件
Merge remote-tracking branch 'origin/dygraph' into dygraph
上级
a2ce5d58
db5d66f9
变更
7
显示空白变更内容
内联
并排
Showing
7 changed file
with
9 addition
and
14 deletion
+9
-14
configs/rec/rec_mtb_nrtr.yml
configs/rec/rec_mtb_nrtr.yml
+1
-1
configs/rec/rec_r45_abinet.yml
configs/rec/rec_r45_abinet.yml
+1
-1
configs/rec/rec_svtrnet.yml
configs/rec/rec_svtrnet.yml
+2
-2
configs/rec/rec_vitstr_none_ce.yml
configs/rec/rec_vitstr_none_ce.yml
+1
-1
ppocr/modeling/heads/rec_abinet_head.py
ppocr/modeling/heads/rec_abinet_head.py
+2
-1
test_tipc/configs/det_r50_db_plusplus/train_infer_python.txt
test_tipc/configs/det_r50_db_plusplus/train_infer_python.txt
+0
-6
test_tipc/docs/benchmark_train.md
test_tipc/docs/benchmark_train.md
+2
-2
未找到文件。
configs/rec/rec_mtb_nrtr.yml
浏览文件 @
69df224c
...
@@ -82,7 +82,7 @@ Train:
...
@@ -82,7 +82,7 @@ Train:
Eval
:
Eval
:
dataset
:
dataset
:
name
:
LMDBDataSet
name
:
LMDBDataSet
data_dir
:
./train_data/data_lmdb_release/
validati
on/
data_dir
:
./train_data/data_lmdb_release/
evaluait
on/
transforms
:
transforms
:
-
DecodeImage
:
# load image
-
DecodeImage
:
# load image
img_mode
:
BGR
img_mode
:
BGR
...
...
configs/rec/rec_r45_abinet.yml
浏览文件 @
69df224c
...
@@ -82,7 +82,7 @@ Train:
...
@@ -82,7 +82,7 @@ Train:
Eval
:
Eval
:
dataset
:
dataset
:
name
:
LMDBDataSet
name
:
LMDBDataSet
data_dir
:
./train_data/data_lmdb_release/
valid
ation/
data_dir
:
./train_data/data_lmdb_release/
evalu
ation/
transforms
:
transforms
:
-
DecodeImage
:
# load image
-
DecodeImage
:
# load image
img_mode
:
RGB
img_mode
:
RGB
...
...
configs/rec/rec_svtrnet.yml
浏览文件 @
69df224c
...
@@ -77,7 +77,7 @@ Metric:
...
@@ -77,7 +77,7 @@ Metric:
Train
:
Train
:
dataset
:
dataset
:
name
:
LMDBDataSet
name
:
LMDBDataSet
data_dir
:
./train_data/data_lmdb_release/training
data_dir
:
./train_data/data_lmdb_release/training
/
transforms
:
transforms
:
-
DecodeImage
:
# load image
-
DecodeImage
:
# load image
img_mode
:
BGR
img_mode
:
BGR
...
@@ -97,7 +97,7 @@ Train:
...
@@ -97,7 +97,7 @@ Train:
Eval
:
Eval
:
dataset
:
dataset
:
name
:
LMDBDataSet
name
:
LMDBDataSet
data_dir
:
./train_data/data_lmdb_release/
validation
data_dir
:
./train_data/data_lmdb_release/
evaluation/
transforms
:
transforms
:
-
DecodeImage
:
# load image
-
DecodeImage
:
# load image
img_mode
:
BGR
img_mode
:
BGR
...
...
configs/rec/rec_vitstr_none_ce.yml
浏览文件 @
69df224c
...
@@ -81,7 +81,7 @@ Train:
...
@@ -81,7 +81,7 @@ Train:
Eval
:
Eval
:
dataset
:
dataset
:
name
:
LMDBDataSet
name
:
LMDBDataSet
data_dir
:
./train_data/data_lmdb_release/
validati
on/
data_dir
:
./train_data/data_lmdb_release/
evaluait
on/
transforms
:
transforms
:
-
DecodeImage
:
# load image
-
DecodeImage
:
# load image
img_mode
:
BGR
img_mode
:
BGR
...
...
ppocr/modeling/heads/rec_abinet_head.py
浏览文件 @
69df224c
...
@@ -273,7 +273,8 @@ def _get_length(logit):
...
@@ -273,7 +273,8 @@ def _get_length(logit):
out
=
out
.
cast
(
'int32'
)
out
=
out
.
cast
(
'int32'
)
out
=
out
.
argmax
(
-
1
)
out
=
out
.
argmax
(
-
1
)
out
=
out
+
1
out
=
out
+
1
out
=
paddle
.
where
(
abn
,
out
,
paddle
.
to_tensor
(
logit
.
shape
[
1
]))
len_seq
=
paddle
.
zeros_like
(
out
)
+
logit
.
shape
[
1
]
out
=
paddle
.
where
(
abn
,
out
,
len_seq
)
return
out
return
out
...
...
test_tipc/configs/det_r50_db_plusplus/train_infer_python.txt
浏览文件 @
69df224c
...
@@ -51,9 +51,3 @@ null:null
...
@@ -51,9 +51,3 @@ null:null
null:null
null:null
===========================infer_benchmark_params==========================
===========================infer_benchmark_params==========================
random_infer_input:[{float32,[3,640,640]}];[{float32,[3,960,960]}]
random_infer_input:[{float32,[3,640,640]}];[{float32,[3,960,960]}]
===========================train_benchmark_params==========================
batch_size:8|16
fp_items:fp32|fp16
epoch:2
--profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096
test_tipc/docs/benchmark_train.md
浏览文件 @
69df224c
...
@@ -9,7 +9,7 @@
...
@@ -9,7 +9,7 @@
```
shell
```
shell
# 运行格式:bash test_tipc/prepare.sh train_benchmark.txt mode
# 运行格式:bash test_tipc/prepare.sh train_benchmark.txt mode
bash test_tipc/prepare.sh test_tipc/configs/det_mv3_db_v2_0/train_
benchmark
.txt benchmark_train
bash test_tipc/prepare.sh test_tipc/configs/det_mv3_db_v2_0/train_
infer_python
.txt benchmark_train
```
```
## 1.2 功能测试
## 1.2 功能测试
...
@@ -33,7 +33,7 @@ dynamic_bs8_fp32_DP_N1C1为test_tipc/benchmark_train.sh传入的参数,格式
...
@@ -33,7 +33,7 @@ dynamic_bs8_fp32_DP_N1C1为test_tipc/benchmark_train.sh传入的参数,格式
## 2. 日志输出
## 2. 日志输出
运行后将保存模型的训练日志和解析日志,使用
`test_tipc/configs/det_mv3_db_v2_0/train_
benchmark
.txt`
参数文件的训练日志解析结果是:
运行后将保存模型的训练日志和解析日志,使用
`test_tipc/configs/det_mv3_db_v2_0/train_
infer_python
.txt`
参数文件的训练日志解析结果是:
```
```
{"model_branch": "dygaph", "model_commit": "7c39a1996b19087737c05d883fd346d2f39dbcc0", "model_name": "det_mv3_db_v2_0_bs8_fp32_SingleP_DP", "batch_size": 8, "fp_item": "fp32", "run_process_type": "SingleP", "run_mode": "DP", "convergence_value": "5.413110", "convergence_key": "loss:", "ips": 19.333, "speed_unit": "samples/s", "device_num": "N1C1", "model_run_time": "0", "frame_commit": "8cc09552473b842c651ead3b9848d41827a3dbab", "frame_version": "0.0.0"}
{"model_branch": "dygaph", "model_commit": "7c39a1996b19087737c05d883fd346d2f39dbcc0", "model_name": "det_mv3_db_v2_0_bs8_fp32_SingleP_DP", "batch_size": 8, "fp_item": "fp32", "run_process_type": "SingleP", "run_mode": "DP", "convergence_value": "5.413110", "convergence_key": "loss:", "ips": 19.333, "speed_unit": "samples/s", "device_num": "N1C1", "model_run_time": "0", "frame_commit": "8cc09552473b842c651ead3b9848d41827a3dbab", "frame_version": "0.0.0"}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录