Predicting film genres from their posters with Tensorflow.js
Find a file
2020-09-18 18:28:16 +01:00
gnuplot GNUplot: change font size 2020-09-18 18:28:16 +01:00
images Move some files around 2020-09-18 17:23:01 +01:00
src FilmPredictor: Fixed prediction error 2020-09-15 19:18:20 +01:00
.gitignore Move some files around 2020-09-18 17:23:01 +01:00
Film Poster Genres - Slides.odp Start working on the slide deck, but it's not finished yet 2020-09-17 17:29:48 +01:00
LICENSE Initial commit 2020-09-15 15:39:02 +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 Update README with a note or 2 2020-09-15 19:12:03 +01:00

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)