Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Just_Paranoid
CnOCR
提交
b4ded5ae
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,发现更多精彩内容 >>
提交
b4ded5ae
编写于
4月 21, 2020
作者:
B
breezedeus
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
no big deal
上级
3b16f2a9
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
17 addition
and
12 deletion
+17
-12
tests/test_mxnet.py
tests/test_mxnet.py
+17
-12
未找到文件。
tests/test_mxnet.py
浏览文件 @
b4ded5ae
...
...
@@ -6,7 +6,6 @@ import mxnet as mx
import
numpy
as
np
from
mxnet
import
nd
import
pytest
from
mxnet.image
import
ImageIter
sys
.
path
.
insert
(
0
,
os
.
path
.
dirname
(
os
.
path
.
dirname
(
os
.
path
.
abspath
(
__file__
))))
sys
.
path
.
insert
(
1
,
os
.
path
.
dirname
(
os
.
path
.
abspath
(
__file__
)))
...
...
@@ -14,7 +13,7 @@ sys.path.insert(1, os.path.dirname(os.path.abspath(__file__)))
from
cnocr.data_utils.aug
import
FgBgFlipAug
from
cnocr.data_utils.data_iter
import
GrayImageIter
LST_DIR
=
Path
(
'data/lst'
)
LST_DIR
=
Path
(
'data/
sample-data-
lst'
)
DATA_DIR
=
Path
(
'data/sample-data'
)
...
...
@@ -39,9 +38,6 @@ def _read_lst_file(fp):
'fp_prefix'
,
[
LST_DIR
/
'sample-data_test'
,
# LST_DIR / 'sample-data_0_train',
# LST_DIR / 'sample-data_1_train',
# LST_DIR / 'sample-data_2_train',
],
)
def
test_iter
(
fp_prefix
):
...
...
@@ -53,18 +49,18 @@ def test_iter(fp_prefix):
rand_mirror
=
False
,
mean
=
None
,
std
=
None
,
brightness
=
0.0
5
,
contrast
=
0.1
,
saturation
=
0.
3
,
hue
=
0.
1
,
pca_noise
=
0.
3
,
brightness
=
0.0
01
,
contrast
=
0.
00
1
,
saturation
=
0.
001
,
hue
=
0.
05
,
pca_noise
=
0.
1
,
inter_method
=
2
,
)
augs
.
append
(
FgBgFlipAug
(
p
=
0.2
))
data_iter
=
GrayImageIter
(
batch_size
=
2
,
data_shape
=
(
3
,
32
,
280
),
label_width
=
1
0
,
label_width
=
2
0
,
path_imgrec
=
str
(
fp_prefix
)
+
".rec"
,
path_imgidx
=
str
(
fp_prefix
)
+
".idx"
,
aug_list
=
augs
,
...
...
@@ -84,6 +80,7 @@ def test_iter(fp_prefix):
# data是一个NDArray,表示第一个batch中的数据,因为这里的batch_size大小是4,所以data的size是2*3*32*280
data
=
batch
.
data
[
0
]
# shape of each one: (3, 32, 280)
# import pdb; pdb.set_trace()
from
matplotlib
import
pyplot
as
plt
...
...
@@ -100,9 +97,17 @@ def test_iter(fp_prefix):
# nd.abs(data[i].astype(np.uint8) - expected_imgs[i].transpose((2, 0, 1)))
# )
# )
print
(
float
(
data
[
i
].
min
()),
float
(
data
[
i
].
max
()))
#
print(float(data[i].min()), float(data[i].max()))
new_img
=
data
[
i
].
asnumpy
()
*
255
plt
.
imshow
(
new_img
.
astype
(
np
.
uint8
).
squeeze
(
axis
=
0
),
cmap
=
'gray'
)
import
cv2
cv2
.
imwrite
(
f
'new-
{
i
}
.png'
,
new_img
.
astype
(
np
.
uint8
).
squeeze
(
axis
=
0
))
plt
.
subplot
(
4
,
1
,
i
*
2
+
2
)
plt
.
imshow
(
expected_imgs
[
i
].
asnumpy
())
plt
.
show
()
def
test_lr_scheduler
():
from
mxnet
import
lr_scheduler
,
optimizer
scheduler
=
lr_scheduler
.
FactorScheduler
(
base_lr
=
1
,
step
=
250
,
factor
=
0.5
)
optim
=
optimizer
.
SGD
(
learning_rate
=
0.1
,
lr_scheduler
=
scheduler
)
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录