https://github.com/himanshuraj98/book-genre-classification

This GitHub repository contains code for a book genre classification project using the VGG16 deep learning model. The project aims to classify books into various genres based on their cover images. The dataset used for training and testing the model consists of 30 book genres.
https://github.com/himanshuraj98/book-genre-classification

classification computer-vision csv deep-learning image image-processing imageclassification ipython-notebook kaggles matlab pytorch rendering transfer-learning vgg16

Last synced: 4 days ago
JSON representation

This GitHub repository contains code for a book genre classification project using the VGG16 deep learning model. The project aims to classify books into various genres based on their cover images. The dataset used for training and testing the model consists of 30 book genres.

• Host: GitHub
• URL: https://github.com/himanshuraj98/book-genre-classification
• Owner: HimanshuRaj98
• License: mit
• Created: 2023-07-25T17:08:55.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2023-07-25T17:08:59.000Z (over 1 year ago)
• Last Synced: 2024-11-13T00:43:34.244Z (2 months ago)
• Topics: classification, computer-vision, csv, deep-learning, image, image-processing, imageclassification, ipython-notebook, kaggles, matlab, pytorch, rendering, transfer-learning, vgg16
• Language: Jupyter Notebook
• Homepage:
• Size: 392 KB
• Stars: 1
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
## Book Genre Classification using VGG16

This GitHub repository contains code for a book genre classification project using the VGG16 deep learning model. The project aims to classify books into various genres based on their cover images. The dataset used for training and testing the model consists of 30 book genres.

Project Structure

## The repository is organized as follows:

1.    cat_to_name.json: A JSON file that maps genre labels to their corresponding names.

2.    checkpoints: A directory to store the trained model checkpoints.

3.    CSV: A directory containing the CSV files for data processing (not visible in this repository).

4.    data_preprocessing.ipynb: A Jupyter Notebook file for preprocessing the data and creating label directories.

5.    book_genre_classification.ipynb: The main Jupyter Notebook file for training the VGG16 model and testing it on the test dataset.

6.    vgg16_30.pth: A saved model checkpoint file for the trained VGG16 model.

## Model Performance

The VGG16 model achieved an accuracy of approximately 31.6% on the test dataset. The top-3 accuracy was approximately 51.8%, and the top-5 accuracy was approximately 63.6%.