Predicting film genres from their posters with Tensorflow.js

Find a file

Starbeamrainbowlabs 180a72df9f Start working on the slide deck, but it's not finished yet		2020-09-17 17:29:48 +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
AI quickstart slide deck diagrams.drawio	Start working on the slide deck, but it's not finished yet	2020-09-17 17:29:48 +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

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)