Multi label Image Classification using PyTorch
The aim of the project is to predict different genres of a movie simply by its poster. Since a movie can have more than 1 genres, this is a multi label image classification problem.
For this assignment, the dataset we’ll be using contains the poster images of several multi-genre movies.
Wei-Ta Chu and Hung-Jui Guo, “Movie Genre Classification based on Poster Images with Deep Neural Networks,” Proceedings of International Workshop on Multimodal Understanding of Social, Affective and Subjective Attributes, pp. 39-45, 2017. (in conjunction with ACM Multimedia 2017)
https://www.cs.ccu.edu.tw/~wtchu/projects/MoviePoster/index.html
The structured and the modified dataset can be downloaded from here
https://drive.google.com/file/d/1iQV5kKF_KGZL9ALx9MMXk_Lg7PklBLCE/view
First, import all the required Python libraries:
import pandas as pd
import numpy as np
from tqdm import tqdm
import os
import matplotlib.pyplot as plt
%matplotlib inline
import torchvision
from torchvision import transforms
from torchvision.utils import make_grid
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch import optim
from torch.utils.data import Dataset
from PIL import Image
!pip install jovian --upgrade -q
import jovian
project_name = 'zerotogans-project'
WARNING: You are using pip version 20.3.1; however, version 20.3.3 is available.
You should consider upgrading via the '/opt/conda/bin/python3.7 -m pip install --upgrade pip' command.