Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
OpenCV
opencv_extra
提交
81c2c97a
O
opencv_extra
项目概览
OpenCV
/
opencv_extra
12 个月 前同步成功
通知
119
Star
882
Fork
1584
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
O
opencv_extra
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
81c2c97a
编写于
7月 11, 2022
作者:
Z
Zihao Mu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
update the test case of Div
上级
fdba14ba
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
53 addition
and
1 deletion
+53
-1
testdata/dnn/onnx/data/input_div_test_1x1_0.npy
testdata/dnn/onnx/data/input_div_test_1x1_0.npy
+0
-0
testdata/dnn/onnx/data/input_div_test_1x1_1.npy
testdata/dnn/onnx/data/input_div_test_1x1_1.npy
+0
-0
testdata/dnn/onnx/data/output_div_test_1x1.npy
testdata/dnn/onnx/data/output_div_test_1x1.npy
+0
-0
testdata/dnn/onnx/generate_onnx_models.py
testdata/dnn/onnx/generate_onnx_models.py
+37
-1
testdata/dnn/onnx/models/div_test_1x1.onnx
testdata/dnn/onnx/models/div_test_1x1.onnx
+16
-0
未找到文件。
testdata/dnn/onnx/data/input_div_test_1x1_0.npy
0 → 100644
浏览文件 @
81c2c97a
文件已添加
testdata/dnn/onnx/data/input_div_test_1x1_1.npy
0 → 100644
浏览文件 @
81c2c97a
文件已添加
testdata/dnn/onnx/data/output_div_test_1x1.npy
0 → 100644
浏览文件 @
81c2c97a
文件已添加
testdata/dnn/onnx/generate_onnx_models.py
浏览文件 @
81c2c97a
...
...
@@ -90,6 +90,23 @@ def save_data_and_onnx_model(name, input_np, output_np, onnx_model):
with
open
(
models_files
,
'wb'
)
as
file
:
file
.
write
(
model_def
.
SerializeToString
())
def
save_data_and_onnx_model_multy_inputs
(
name
,
input_list
,
output_np
,
onnx_model
):
for
index
in
range
(
len
(
input_list
)):
print
(
name
+
" input "
+
str
(
index
)
+
" has sizes"
,
input_list
[
index
].
shape
)
input_files
=
os
.
path
.
join
(
"data"
,
"input_"
+
name
+
"_"
+
str
(
index
))
np
.
save
(
input_files
,
input_list
[
index
])
print
(
name
+
" output has sizes"
,
output_np
.
shape
)
print
()
output_files
=
os
.
path
.
join
(
"data"
,
"output_"
+
name
)
np
.
save
(
output_files
,
np
.
ascontiguousarray
(
output_np
.
data
))
models_files
=
os
.
path
.
join
(
"models"
,
name
+
".onnx"
)
onnx_model_pb
=
onnx
.
_serialize
(
onnx_model
)
model_def
=
assertONNXExpected
(
onnx_model_pb
)
with
open
(
models_files
,
'wb'
)
as
file
:
file
.
write
(
model_def
.
SerializeToString
())
def
simplify
(
name
,
rename
=
False
,
**
kwargs
):
model
,
check
=
onnxsim
.
simplify
(
name
,
**
kwargs
)
...
...
@@ -1725,4 +1742,23 @@ graph2 = onnx.helper.make_graph(nodes2,
outputs
,
initializer
=
[
weight_tensor
])
gemm_model2
=
onnx
.
helper
.
make_model
(
graph2
)
output_np
=
gemm_reference_implementation
(
input_np
,
weight_np
)
save_data_and_onnx_model
(
"gemm_transB_0"
,
input_np
,
output_np
,
gemm_model2
)
\ No newline at end of file
save_data_and_onnx_model
(
"gemm_transB_0"
,
input_np
,
output_np
,
gemm_model2
)
# ########################## DivBroadcast ##########################
input_np
=
np
.
random
.
rand
(
1
,
4
).
astype
(
"float32"
)
input2_np
=
np
.
random
.
rand
(
1
,
1
).
astype
(
np
.
float32
)
inputs
=
[
onnx
.
helper
.
make_tensor_value_info
(
"input1"
,
onnx
.
mapping
.
NP_TYPE_TO_TENSOR_TYPE
[
input_np
.
dtype
],
shape
=
input_np
.
shape
),
\
onnx
.
helper
.
make_tensor_value_info
(
"input2"
,
onnx
.
mapping
.
NP_TYPE_TO_TENSOR_TYPE
[
input2_np
.
dtype
],
shape
=
input2_np
.
shape
)]
outputs
=
[
onnx
.
helper
.
make_tensor_value_info
(
"output"
,
onnx
.
TensorProto
.
FLOAT
,
shape
=
(
1
,
4
))]
nodes
=
[
onnx
.
helper
.
make_node
(
"Div"
,
[
"input1"
,
"input2"
],
[
"output"
])]
graph
=
onnx
.
helper
.
make_graph
(
nodes
,
"div_test"
,
inputs
,
outputs
)
onnx_model
=
onnx
.
helper
.
make_model
(
graph
)
output_np
=
input_np
/
input2_np
save_data_and_onnx_model_multy_inputs
(
"div_test_1x1"
,
[
input_np
,
input2_np
],
output_np
,
onnx_model
)
\ No newline at end of file
testdata/dnn/onnx/models/div_test_1x1.onnx
0 → 100644
浏览文件 @
81c2c97a
:w
input1
input2output"Divdiv_testZ
input1
Z
input2
b
output
B
\ No newline at end of file
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录