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.

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 a TensorFlow.js model to do inference #43

Merged
merged 21 commits into from
Nov 1, 2018
Merged

Conversation

mrbarbasa
Copy link
Contributor

@mrbarbasa mrbarbasa commented Oct 30, 2018

Got TensorFlow.js to run on Node and make predictions based on the 1000 classes of ImageNet. Sample output when you run node server/model.js (be sure to npm install first, have Node v10.12.0, and have Python v2.7.x):

Identification Results: [ { className: 'brain coral', probability: 0.3537396490573883 },
  { className: 'coral fungus', probability: 0.1802905946969986 },
  { className: 'starfish, sea star',
    probability: 0.14114698767662048 } ]

See this blog post if you'd like to learn more: http://jamesthom.as/blog/2018/08/07/machine-learning-in-node-dot-js-with-tensorflow-dot-js/

It's not yet robust and does not work on the ohia picture (in /server/images). I'm continuing to work off of this to generalize it and tie it to the frontend. Also, I'd like to try running an equivalent version in the browser for comparison. -- EDIT: Going with server, no time for trying it out on the browser.

@mrbarbasa mrbarbasa self-assigned this Oct 30, 2018
@mrbarbasa mrbarbasa added the in progress Work is in progress label Oct 30, 2018
@mrbarbasa mrbarbasa removed the in progress Work is in progress label Nov 1, 2018
@mrbarbasa mrbarbasa merged commit a4cdc26 into master Nov 1, 2018
@mrbarbasa mrbarbasa deleted the feature/tensorflowjs branch November 1, 2018 04:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Refine the image upload process to S3 Create TensorFlow JS model to do inference
1 participant