Skip to content

Predicted the genre of the book from its cover using convolution neural networks.

Notifications You must be signed in to change notification settings

sayaliwalke30/JudgeBookByCover

Repository files navigation

JudgeBookByCover

Problem Statement: The goal is to determine if genre information can be learned based on the visual aspects of a cover created by the designer. 

/**

author Sayali Walke

**/

Solution:

To tackle this problem I used the concept of transfer learning and developed a Convolutional Neural Networks (CNN) based system for book cover genre classification. I used a pre-trained network on ImageNet. By pre-training VGG16 and ResNet50 models on a very large dataset such as ImageNet, its possible to take advantage of the learned features and transfer it to other applications.

Steps:

1] Web scraped book cover images from Amazon.com
2] Data exploration
3] CNN from scratch
4] Transfer learning using VGG16 and ResNet50

Releases

No releases published

Packages

No packages published