404 B
404 B
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)