-
Notifications
You must be signed in to change notification settings - Fork 152
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add ncnn deployment examples #145
Conversation
Codecov Report
@@ Coverage Diff @@
## master #145 +/- ##
==========================================
+ Coverage 84.34% 84.49% +0.15%
==========================================
Files 11 12 +1
Lines 811 819 +8
==========================================
+ Hits 684 692 +8
Misses 127 127
Flags with carried forward coverage won't be shown. Click here to find out more.
Continue to review full report at Codecov.
|
e333d9e
to
77f295a
Compare
Reshape Reshape_259 1 1 671 689 0=-1 1=85 2=3 | ||
Permute Transpose_260 1 1 689 h1 0=1 | ||
Convolution Conv_261 1 1 644_splitncnn_0 691 0=255 1=1 5=1 6=65280 | ||
Reshape Reshape_275 1 1 691 709 0=-1 1=85 2=3 | ||
Permute Transpose_276 1 1 709 h2 0=1 | ||
Convolution Conv_277 1 1 670 711 0=255 1=1 5=1 6=130560 | ||
Reshape Reshape_291 1 1 711 729 0=-1 1=85 2=3 | ||
Permute Transpose_292 1 1 729 h3 0=1 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Exported natively by ncnnoptimize
is
Reshape Reshape_259 1 1 671 689 0=6400 1=-1 2=3
Permute Transpose_260 1 1 689 h1 0=1
Convolution Conv_261 1 1 644_splitncnn_0 691 0=255 1=1 5=1 6=65280
Reshape Reshape_275 1 1 691 709 0=1600 1=-1 2=3
Permute Transpose_276 1 1 709 h2 0=1
Convolution Conv_277 1 1 670 711 0=255 1=1 5=1 6=130560
Reshape Reshape_291 1 1 711 729 0=400 1=-1 2=3
Permute Transpose_292 1 1 729 h3 0=1
7767517 | ||
176 200 | ||
Input images 0 1 images | ||
YOLOv5Focus focus 1 1 images 401 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Exported natively by onnx2ncnn
is
185 212
Input images 0 1 images
Split splitncnn_input0 1 4 images images_splitncnn_0 images_splitncnn_1 images_splitncnn_2 images_splitncnn_3
Crop Slice_4 1 1 images_splitncnn_3 365 -23309=1,0 -23310=1,2147483647 -23311=1,1
Crop Slice_9 1 1 365 370 -23309=1,0 -23310=1,2147483647 -23311=1,2
Crop Slice_14 1 1 images_splitncnn_2 375 -23309=1,1 -23310=1,2147483647 -23311=1,1
Crop Slice_19 1 1 375 380 -23309=1,0 -23310=1,2147483647 -23311=1,2
Crop Slice_24 1 1 images_splitncnn_1 385 -23309=1,0 -23310=1,2147483647 -23311=1,1
Crop Slice_29 1 1 385 390 -23309=1,1 -23310=1,2147483647 -23311=1,2
Crop Slice_34 1 1 images_splitncnn_0 395 -23309=1,1 -23310=1,2147483647 -23311=1,1
Crop Slice_39 1 1 395 400 -23309=1,1 -23310=1,2147483647 -23311=1,2
Concat Concat_40 4 1 370 380 390 400 401 0=0
@@ -170,6 +170,15 @@ def focus_transform(x: Tensor) -> Tensor: | |||
return y | |||
|
|||
|
|||
def space_to_depth(x: Tensor) -> Tensor: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
We don't use this function yet, but we'll put it here for now and see how to call this part later, or do some experiments with it.
cc @xiguadong
The whole process of exporting |
The whole work is just in the beginning of deploying cc @xiguadong . |
ONNX
export toolsncnn
C++ inference examplesncnn
Python inference examples