Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
OpenCV
opencv_extra
提交
2702fbcc
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,发现更多精彩内容 >>
提交
2702fbcc
编写于
1月 31, 2022
作者:
R
rogday
提交者:
Smirnov Egor
2月 03, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add tests
上级
b2e7c8ac
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
41 addition
and
8 deletion
+41
-8
testdata/dnn/onnx/data/input_slice_neg_starts.npy
testdata/dnn/onnx/data/input_slice_neg_starts.npy
+0
-0
testdata/dnn/onnx/data/output_slice_neg_starts.npy
testdata/dnn/onnx/data/output_slice_neg_starts.npy
+0
-0
testdata/dnn/onnx/generate_onnx_models.py
testdata/dnn/onnx/generate_onnx_models.py
+41
-8
testdata/dnn/onnx/models/slice_neg_starts.onnx
testdata/dnn/onnx/models/slice_neg_starts.onnx
+0
-0
未找到文件。
testdata/dnn/onnx/data/input_slice_neg_starts.npy
浏览文件 @
2702fbcc
无法预览此类型文件
testdata/dnn/onnx/data/output_slice_neg_starts.npy
浏览文件 @
2702fbcc
无法预览此类型文件
testdata/dnn/onnx/generate_onnx_models.py
浏览文件 @
2702fbcc
...
...
@@ -434,16 +434,49 @@ model = Slice()
save_data_and_model
(
"slice"
,
input
,
model
)
save_data_and_model
(
"slice_opset_11"
,
input
,
model
,
version
=
11
)
class
SliceStarts
(
nn
.
Module
):
def
__init__
(
self
,
*
args
,
**
kwargs
):
super
(
SliceStarts
,
self
).
__init__
()
def
generate_slice_neg_starts
(
):
x
=
np
.
random
.
randn
(
2
,
3
,
4
,
3
).
astype
(
np
.
float32
)
y
=
x
[
-
1
:
2
,
-
3
:
-
1
,
2
:
3
,
1
:
-
1
]
def
forward
(
self
,
x
):
return
x
[
-
1
:]
starts
=
np
.
array
([
-
1
,
-
3
,
2
,
1
],
dtype
=
np
.
int64
)
starts
=
onnx
.
numpy_helper
.
from_array
(
starts
,
name
=
'starts'
)
ends
=
np
.
array
([
2
,
-
1
,
3
,
-
1
],
dtype
=
np
.
int64
)
ends
=
onnx
.
numpy_helper
.
from_array
(
ends
,
name
=
'ends'
)
model
=
SliceStarts
()
input_
=
Variable
(
torch
.
randn
(
1
,
10
,
dtype
=
torch
.
float32
))
save_data_and_model
(
"slice_neg_starts"
,
input_
,
model
)
node
=
onnx
.
helper
.
make_node
(
'Slice'
,
inputs
=
[
'X'
,
'starts'
,
'ends'
],
outputs
=
[
'Y'
],
)
X
=
onnx
.
helper
.
make_tensor_value_info
(
'X'
,
onnx
.
TensorProto
.
FLOAT
,
list
(
x
.
shape
))
Y
=
onnx
.
helper
.
make_tensor_value_info
(
'Y'
,
onnx
.
TensorProto
.
FLOAT
,
list
(
y
.
shape
))
graph
=
onnx
.
helper
.
make_graph
(
[
node
],
# nodes
'slice_neg_starts'
,
# name
[
X
],
# inputs
[
Y
],
# outputs
)
graph
.
initializer
.
append
(
starts
)
graph
.
initializer
.
append
(
ends
)
model
=
onnx
.
helper
.
make_model
(
graph
,
producer_name
=
'onnx'
)
onnx
.
checker
.
check_model
(
model
)
name
=
'slice_neg_starts'
input_files
=
os
.
path
.
join
(
"data"
,
"input_"
+
name
)
np
.
save
(
input_files
,
x
.
data
)
output_files
=
os
.
path
.
join
(
"data"
,
"output_"
+
name
)
np
.
save
(
output_files
,
np
.
ascontiguousarray
(
y
.
data
))
models_files
=
os
.
path
.
join
(
"models"
,
name
+
".onnx"
)
onnx
.
save
(
model
,
models_files
)
generate_slice_neg_starts
()
input_2
=
Variable
(
torch
.
randn
(
6
,
6
))
custom_slice_list
=
[
...
...
testdata/dnn/onnx/models/slice_neg_starts.onnx
浏览文件 @
2702fbcc
无法预览此类型文件
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录