Predicting film genres from their posters with Tensorflow.js
Go to file
Starbeamrainbowlabs c9e7826416
FilmPredictor: Fixed prediction error
2020-09-15 19:18:20 +01:00
src FilmPredictor: Fixed prediction error 2020-09-15 19:18:20 +01:00
.gitignore It trains! 2020-09-15 18:32:51 +01:00
LICENSE Initial commit 2020-09-15 15:39:02 +01:00
README.md Update README with a note or 2 2020-09-15 19:12:03 +01:00
package-lock.json It trains! 2020-09-15 18:32:51 +01:00
package.json It trains! 2020-09-15 18:32:51 +01:00

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)