Predicting film genres from their posters with Tensorflow.js
Go to file
Starbeamrainbowlabs 304aa66358
Update README with a note or 2
....we'll need to fill it out, but that can wait until tomorrow :P
2020-09-15 19:12:03 +01:00
src It predicts! 2020-09-15 19:09:42 +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)