Skip to content

Using movie trailers to identify Genre | Deep Learning for Video Classification

Notifications You must be signed in to change notification settings

maximus009/MovieScope

Repository files navigation

MovieScope

Please refer to the other repo - https://github.com/maximus009/VisLangResearch

Idea is to extend video classification for identifying what the genre (romance, horror or action) of a movie is based on the trailer solely using visual features.
Currently, using only spatial CNN features (VGG), to extract features for every frame extracted.
Frame extraction can be a challenge in itself. Currently, sampling frame at a fixed timestep. Can also implement shot-boundary detection, and split the trailers into scenes which might (or not) be useful in training.

MovieScope is certainly, started with movie trailers, which are long videos compared to other videos used for activity classification and recognition. Will need to use recurrent networks and LSTMs as well Refer to branch "lstm" to see the implementation.


To run the code, first, make changes to the file config/resources.py with the appropriate directory.

Steps to run the program

  1. create_data.py ["need to run twice, one for train data, one for test data; will refine the Code"]
  2. create_model.py
  3. test_model.py

The program is pretty easy to use, plus the repo contains other auxilliary codes too.

This project was initiated as a class project for Computational Visual Recognition class at UVa (Fall 2016). The project is still under development.

Please contact [email protected] for any clarifications or questions.


About

Using movie trailers to identify Genre | Deep Learning for Video Classification

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages