Skip to content

AlexanderWangY/GuessTheGenre

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 

Repository files navigation

Movie Genre Guessing ML Model (@ Resnet152 & FastAI + Pytorch)

This is a computer vision classification model trained using FastAI and PyTorch using the resnet152 framework. It classifies between 5 genres viz: Horror, Action, Fantasy, Animation, and Romance.

The training process included: Preprocessing -> Training -> Cleaning the data -> Training -> Export

You can test it @ HuggingFace Space

About

A Computer Vision Model to guess the genre of movie posters! Deployed on HuggingFace Spaces.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published