Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Just_Paranoid
CnOCR
提交
826a92d8
CnOCR
项目概览
Just_Paranoid
/
CnOCR
与 Fork 源项目一致
Fork自
Cloud IDE / CnOCR
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
CnOCR
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
826a92d8
编写于
5月 04, 2020
作者:
B
breezedeus
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
use param `gpu` at cmd `cnocr_evaluate`
上级
8f41bdc4
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
20 addition
and
12 deletion
+20
-12
cnocr/fit/fit.py
cnocr/fit/fit.py
+4
-5
cnocr/utils.py
cnocr/utils.py
+9
-0
scripts/cnocr_evaluate.py
scripts/cnocr_evaluate.py
+7
-7
未找到文件。
cnocr/fit/fit.py
浏览文件 @
826a92d8
...
...
@@ -2,6 +2,8 @@ import logging
import
os
import
mxnet
as
mx
from
..utils
import
gen_context
def
_load_model
(
args
):
if
'load_epoch'
not
in
args
or
args
.
load_epoch
is
None
:
...
...
@@ -16,10 +18,7 @@ def _load_model(args):
def
fit
(
network
,
data_train
,
data_val
,
metrics
,
args
,
hp
,
data_names
=
None
):
if
args
.
gpu
>
0
:
contexts
=
[
mx
.
context
.
gpu
(
i
)
for
i
in
range
(
args
.
gpu
)]
else
:
contexts
=
[
mx
.
context
.
cpu
()]
context
=
gen_context
(
args
.
gpu
)
logging
.
info
(
'hp: %s'
,
hp
)
sym
,
arg_params
,
aux_params
=
_load_model
(
args
)
...
...
@@ -32,7 +31,7 @@ def fit(network, data_train, data_val, metrics, args, hp, data_names=None):
symbol
=
network
,
data_names
=
[
"data"
]
if
data_names
is
None
else
data_names
,
label_names
=
[
'label'
],
context
=
context
s
,
context
=
context
,
)
begin_epoch
=
args
.
load_epoch
if
args
.
load_epoch
else
0
...
...
cnocr/utils.py
浏览文件 @
826a92d8
...
...
@@ -20,6 +20,7 @@ from pathlib import Path
import
logging
import
platform
import
zipfile
import
mxnet
as
mx
from
mxnet.gluon.utils
import
download
from
.consts
import
AVAILABLE_MODELS
,
EMB_MODEL_TYPES
,
SEQ_MODEL_TYPES
...
...
@@ -55,6 +56,14 @@ def set_logger(log_file=None, log_level=logging.INFO, log_file_level=logging.NOT
return
logger
def
gen_context
(
num_gpu
):
if
num_gpu
>
0
:
context
=
[
mx
.
context
.
gpu
(
i
)
for
i
in
range
(
num_gpu
)]
else
:
context
=
[
mx
.
context
.
cpu
()]
return
context
def
data_dir_default
():
"""
...
...
scripts/cnocr_evaluate.py
浏览文件 @
826a92d8
...
...
@@ -33,7 +33,7 @@ import Levenshtein
sys
.
path
.
insert
(
0
,
os
.
path
.
dirname
(
os
.
path
.
dirname
(
os
.
path
.
abspath
(
__file__
))))
from
cnocr
import
CnOcr
from
cnocr.utils
import
set_logger
from
cnocr.utils
import
set_logger
,
gen_context
logger
=
set_logger
(
log_level
=
logging
.
INFO
)
...
...
@@ -46,11 +46,10 @@ def evaluate():
)
parser
.
add_argument
(
"--model-epoch"
,
type
=
int
,
default
=
None
,
help
=
"model epoch"
)
parser
.
add_argument
(
"--context"
,
type
=
str
,
default
=
'cpu'
,
choices
=
[
'cpu'
,
'gpu'
],
help
=
"which context to run inferences"
,
"--gpu"
,
help
=
"Number of GPUs for training [Default 0, means using cpu]"
,
type
=
int
,
default
=
0
,
)
parser
.
add_argument
(
"-i"
,
...
...
@@ -76,9 +75,10 @@ def evaluate():
help
=
"the output directory which records the analysis results"
,
)
args
=
parser
.
parse_args
()
context
=
gen_context
(
args
.
gpu
)
ocr
=
CnOcr
(
model_name
=
args
.
model_name
,
model_epoch
=
args
.
model_epoch
,
context
=
args
.
context
model_name
=
args
.
model_name
,
model_epoch
=
args
.
model_epoch
,
context
=
context
)
alphabet
=
ocr
.
_alphabet
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录