Skip to content

Commit

Permalink
Add a function of pose and object detection for offline
Browse files Browse the repository at this point in the history
  • Loading branch information
JeongJun-Lee committed Aug 27, 2022
1 parent f5dd56e commit bcb8b12
Show file tree
Hide file tree
Showing 18 changed files with 21,557 additions and 1 deletion.
4 changes: 3 additions & 1 deletion package.json
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,8 @@
"@tensorflow-models/coco-ssd": "^2.0.1",
"@tensorflow-models/posenet": "^2.2.1",
"@tensorflow/tfjs": "^1.5.2",
"@tensorflow/tfjs-backend-cpu": "^3.20.0",
"@tensorflow/tfjs-backend-webgl": "^3.20.0",
"@tensorflow/tfjs-converter": "^1.5.2",
"@tensorflow/tfjs-core": "^1.5.2",
"audiobuffer-to-wav": "^1.0.0",
Expand All @@ -53,8 +55,8 @@
"isomorphic-fetch": "^2.2.1",
"jest": "^24.9.0",
"lodash": "^4.17.15",
"pixi.js": "^5.2.1",
"mathjs": "^7.1.0",
"pixi.js": "^5.2.1",
"postcss-flexbugs-fixes": "^4.1.0",
"simplebar": "^3.1.1",
"skmeans": "^0.11.3",
Expand Down
2 changes: 2 additions & 0 deletions src/util/workers/video.worker.ts
Original file line number Diff line number Diff line change
Expand Up @@ -180,6 +180,7 @@ ctx.onmessage = async function(e: {
outputStride: 16,
inputResolution: dimension,
multiplier: 0.5,
modelUrl: weightsUrl+"/posenet/model-stride16.json"
})
.then((mobileNetLoaded: any) => {
mobileNet = mobileNetLoaded;
Expand All @@ -189,6 +190,7 @@ ctx.onmessage = async function(e: {
cocoSsd
.load({
base: 'lite_mobilenet_v2',
modelUrl: weightsUrl+"/coco-ssd/model.json"
})
.then((cocoLoaded: any) => {
coco = cocoLoaded;
Expand Down
Binary file added weights/coco-ssd/group1-shard1of5
Binary file not shown.
Binary file added weights/coco-ssd/group1-shard2of5
Binary file not shown.
Binary file added weights/coco-ssd/group1-shard3of5
Binary file not shown.
Binary file added weights/coco-ssd/group1-shard4of5
Binary file not shown.
Binary file added weights/coco-ssd/group1-shard5of5
Binary file not shown.
Loading

0 comments on commit bcb8b12

Please sign in to comment.