Skip to content

Commit

Permalink
modified for cpu, uses tfjs-node
Browse files Browse the repository at this point in the history
  • Loading branch information
pangyuteng committed Jul 3, 2020
1 parent 601a9a5 commit 46a972f
Showing 4 changed files with 3 additions and 8 deletions.
5 changes: 1 addition & 4 deletions bodypix/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,7 +1,4 @@
# Base image with TensorFlow GPU requirements
#FROM nvcr.io/nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04
#FROM nvidia/cuda:10.2-cudnn7-runtime-ubuntu18.04
FROM nvidia/cuda:10.2-runtime-ubuntu18.04
FROM ubuntu:18.04
# Install node
RUN apt update && apt install -y curl make build-essential \
&& curl -sL https://deb.nodesource.com/setup_12.x | bash - \
2 changes: 1 addition & 1 deletion bodypix/app.js
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
const tf = require('@tensorflow/tfjs-node-gpu');
const tf = require('@tensorflow/tfjs-node');
const bodyPix = require('@tensorflow-models/body-pix');
const http = require('http');
(async () => {
2 changes: 1 addition & 1 deletion bodypix/package.json
Original file line number Diff line number Diff line change
@@ -3,6 +3,6 @@
"version": "0.0.1",
"dependencies": {
"@tensorflow-models/body-pix": "^2.0.5",
"@tensorflow/tfjs-node-gpu": "^1.7.1"
"@tensorflow/tfjs-node": "^1.7.1"
}
}
2 changes: 0 additions & 2 deletions docker-compose.yml
Original file line number Diff line number Diff line change
@@ -9,8 +9,6 @@ services:
ports:
- 9000:9000
command: node /src/app.js
devices:
- /dev/nvidia0:/dev/nvidia0
fakecam:
build:
context: ./fakecam

0 comments on commit 46a972f

Please sign in to comment.