Predicting film genres from their posters with Tensorflow.js
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
Starbeamrainbowlabs c9e7826416
FilmPredictor: Fixed prediction error
2 years ago
src FilmPredictor: Fixed prediction error 2 years ago
.gitignore It trains! 2 years ago
LICENSE Initial commit 2 years ago
README.md Update README with a note or 2 2 years ago
package-lock.json It trains! 2 years ago
package.json It trains! 2 years ago

README.md

film-poster-genres

Predicting film genres from their posters with Tensorflow.js

To use:

node src/index.mjs train --input datasets/posters-256/ --output output/
node src/index.mjs predict --input datasets/posters-256/validate/25410,Comedy,Family,Drama.jpg --ai-model output/checkpoints/49 2>/dev/null

TODO: Fill out this README (and maybe move it to GitHub & make this repo a clone)