Image-Captioning

KO | EN

gitlite — search

#hacktoberfest #ai #llm #go #golang #deep-learning #macos #nodejs #windows #awesome #cli #awesome-list #pytorch #openai

Image-Captioning

(★ 11)

Image captioning using CNN and RNN

Image-Captioning Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

### Image Captioning using CNN-RNN Architecture

#### Video tutorial: https://youtu.be/htNmFL2BG34

![img_cap_model_parial](https://github.com/AarohiSingla/Image-Captioning/assets/60029146/3665bf55-22df-4cfa-86f9-9ae3f8ba106d)

The CNN-RNN architecture for image captioning consists of two main components: the CNN encoder and the RNN decoder.

##### CNN Encoder:
The CNN encoder is responsible for extracting features from the input image and then flattened the features. After that we are sending those flattened features to embedding layer the get the embedded features. These embedded features will become the input to decoder.

##### RNN Decoder:
The input to RNN Decoder is the output of CNN encoder. RNN generates a sequence of words that describe the image. It uses a recurrent neural network, such as a Long Short-Term Memory (LSTM) or a Gated Recurrent Unit (GRU), to generate the sequence. The decoder is trained to predict the next word in the sequence given the previous words.

##### This code is tested using Python 3.8 and Pytorch

### Dataset:
To train the CNN-RNN model for image captioning, we need a dataset of images paired with their corresponding captions.

## Credits

This Repo includes the code from the following repository:

- [Original Repository](https://github.com/iamirmasoud/image_captioning)

Special thanks to (https://github.com/iamirmasoud) for their contribution.

Image-Captioning

Image-Captioning Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Image-Captioning

Image-Captioning Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README