/Image-Caption-Generator-CNN

This project Implements a combination of Convolutional Neural Networks (CNNs) and Recurrent Neural Networks (RNNs) to generate descriptive captions for input images.

Primary LanguageJupyter Notebook

Image Caption Generator CNN

Screenshot 2024-01-27 at 01 06 34

Overview

This project implements an Image Caption Generator using a combination of Convolutional Neural Networks (CNNs) and Recurrent Neural Networks (RNNs). The goal is to generate descriptive captions for input images.

Dependencies

  • Python 3.x
  • TensorFlow
  • VisionEncoderDecoderModel
  • ViTFeatureExtractor
  • AutoTokenizer

Installation

  1. Clone the repository: 
    git clone https://github.com/praveenliyanage/image-caption-generator-cnn.git
cd image-caption-generator

Installed Dependancies

- !pip install transformers