Skip to content
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.

Sign up for GitHub

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

Added new models to ModelHub: PoseNet and EfficientDet family #317

Merged
merged 20 commits into from
Jan 12, 2022
+2,372 −102
Merged

Added new models to ModelHub: PoseNet and EfficientDet family #317

Changes from 1 commit
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
70df7ea
Added an initial support for PoseNet
zaleslaw Oct 26, 2021
8f515a1
Added some new images
zaleslaw Oct 27, 2021
3f12e21
Added an Easy API
zaleslaw Oct 28, 2021
03a12a2
Added an edges support
zaleslaw Oct 29, 2021
6a843e2
Fixed bug with the input tensor
zaleslaw Oct 29, 2021
fcfea4b
Fixed edges calculation
zaleslaw Oct 29, 2021
c107568
Added a prototype
zaleslaw Dec 11, 2021
1117203
Added low-level EfficientDet support
zaleslaw Dec 16, 2021
9bd1195
Added an EfficientDet family
zaleslaw Dec 20, 2021
993e733
Added an EfficientDet family
zaleslaw Dec 20, 2021
a952650
Merge branch 'master' into feature-269
zaleslaw Dec 21, 2021
756ba93
Added an EfficientDetObjectDetectionModel
zaleslaw Dec 21, 2021
b5619df
Fixed a visualisation
zaleslaw Dec 21, 2021
31a47d5
Extracted pre-initialization of ONNX models to the separate models
zaleslaw Dec 21, 2021
90949b6
Extended predictRaw method
zaleslaw Dec 22, 2021
224f2d7
Added KDocs
zaleslaw Dec 22, 2021
f843c69
Fixed tests
zaleslaw Dec 22, 2021
00c65c7
Fixed tests
zaleslaw Dec 23, 2021
7381e30
Fixed tests
zaleslaw Dec 27, 2021
18ae1c9
Fixed tests
zaleslaw Dec 27, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fixed tests
@zaleslaw
zaleslaw committed Dec 23, 2021
commit 00c65c71fe31ad556641c4e2a1daee8be56e2882
4 changes: 2 additions & 2 deletions examples/src/test/kotlin/examples/onnx/posedetection/PoseDetectionTestSuite.kt
View file
Original file line number Diff line number Diff line change
@@ -86,7 +86,7 @@ class PoseDetectionTestSuite {

val yhat = it.predictRaw(inputData)

val rawPoseLandMarks = (yhat as List<Array<Array<Array<FloatArray>>>>)[0][0][0]
val rawPoseLandMarks = (yhat["output_0"] as Array<Array<Array<FloatArray>>>)[0][0]

assertEquals(17, rawPoseLandMarks.size)
}
@@ -118,7 +118,7 @@ class PoseDetectionTestSuite {

val yhat = it.predictRaw(inputData)

val rawPoseLandMarks = (yhat as List<Array<Array<Array<FloatArray>>>>)[0][0][0]
val rawPoseLandMarks = (yhat["output_0"] as Array<Array<Array<FloatArray>>>)[0][0]

assertEquals(17, rawPoseLandMarks.size)
}