Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
OpenCV
opencv_extra
提交
b543e775
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,发现更多精彩内容 >>
提交
b543e775
编写于
12月 20, 2021
作者:
R
rogday
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add test data for Div
上级
01e8a30a
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
26 addition
and
0 deletion
+26
-0
testdata/dnn/onnx/data/input_div_const.npy
testdata/dnn/onnx/data/input_div_const.npy
+0
-0
testdata/dnn/onnx/data/output_div_const.npy
testdata/dnn/onnx/data/output_div_const.npy
+0
-0
testdata/dnn/onnx/generate_onnx_models.py
testdata/dnn/onnx/generate_onnx_models.py
+26
-0
testdata/dnn/onnx/models/div_const.onnx
testdata/dnn/onnx/models/div_const.onnx
+0
-0
未找到文件。
testdata/dnn/onnx/data/input_div_const.npy
0 → 100644
浏览文件 @
b543e775
文件已添加
testdata/dnn/onnx/data/output_div_const.npy
0 → 100644
浏览文件 @
b543e775
文件已添加
testdata/dnn/onnx/generate_onnx_models.py
浏览文件 @
b543e775
...
...
@@ -1558,3 +1558,29 @@ class SubFromConst1(nn.Module):
model
=
SubFromConst1
()
input_
=
Variable
(
torch
.
randn
(
1
,
2
,
3
,
4
,
dtype
=
torch
.
float32
))
save_data_and_model
(
"sub_from_const1"
,
input_
,
model
)
########################## const / x ##########################
node
=
onnx
.
helper
.
make_node
(
'Div'
,
inputs
=
[
'x'
,
'y'
],
outputs
=
[
'z'
])
x
=
np
.
array
([
2
]).
astype
(
np
.
float32
)
y
=
np
.
array
([[
4
,
4
],
[
4
,
4
]]).
astype
(
np
.
float32
)
name
=
'div_const'
input_files
=
os
.
path
.
join
(
"data"
,
"input_"
+
name
)
np
.
save
(
input_files
,
x
.
data
)
np
.
save
(
input_files
,
y
.
data
)
z
=
(
x
/
y
).
astype
(
np
.
float32
)
output_files
=
os
.
path
.
join
(
"data"
,
"output_"
+
name
)
np
.
save
(
output_files
,
np
.
ascontiguousarray
(
z
.
data
))
X
=
onnx
.
helper
.
make_tensor
(
'x'
,
onnx
.
TensorProto
.
FLOAT
,
x
.
shape
,
x
)
Y
=
onnx
.
helper
.
make_tensor_value_info
(
'y'
,
onnx
.
TensorProto
.
FLOAT
,
y
.
shape
)
Z
=
onnx
.
helper
.
make_tensor_value_info
(
'z'
,
onnx
.
TensorProto
.
FLOAT
,
z
.
shape
)
graph
=
onnx
.
helper
.
make_graph
([
node
],
'div_const'
,
[
Y
],
[
Z
],
initializer
=
[
X
])
model
=
onnx
.
helper
.
make_model
(
graph
,
producer_name
=
name
)
models_files
=
os
.
path
.
join
(
"models"
,
name
+
".onnx"
)
onnx
.
save
(
model
,
models_files
)
########################## const / x ##########################
testdata/dnn/onnx/models/div_const.onnx
0 → 100644
浏览文件 @
b543e775
文件已添加
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录