brozek/bodypix-background
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Combined into main due to async ordering errors

Browse source
This commit is contained in:
Brandon Rozek 2020-05-03 21:19:41 -04:00
parent 8817c68d87
commit 74ac5a903d
1 changed files with 3 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
server.js
View file

@ -3,18 +3,13 @@ const tf = require('@tensorflow/tfjs-node');
const bodyPix = require('@tensorflow-models/body-pix'); const bodyPix = require('@tensorflow-models/body-pix');
const {decodeJpeg} = require('./decode_image'); const {decodeJpeg} = require('./decode_image');
let net = null; async function main() {
const net = await bodyPix.load({
async function load() {
net = await bodyPix.load({
architecture: 'MobileNetV1', architecture: 'MobileNetV1',
outputStride: 16, outputStride: 16,
multiplier: 0.5, multiplier: 0.5,
quantBytes: 2 quantBytes: 2
}); });
}
async function run() {
const sock = new zmq.Reply; const sock = new zmq.Reply;
await sock.bind('ipc:///tmp/bodypix'); await sock.bind('ipc:///tmp/bodypix');
@ -28,5 +23,4 @@ async function run() {
} }
} }
load(); main();
run();