sureal01 commited on
Commit
ff6048b
·
verified ·
1 Parent(s): 5c4da1f

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -0
README.md CHANGED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - image-captioning
4
+ - deep-learning
5
+ - pytorch
6
+ - encoder-decoder
7
+ - vision
8
+ ---
9
+
10
+ # 🖼️ Image Captioning Model
11
+
12
+ This is a deep learning-based **image captioning model** trained using a **CNN Encoder + LSTM Decoder** architecture. The model generates captions for input images based on visual features extracted by a Convolutional Neural Network (CNN).
13
+
14
+ ## 📌 Model Details
15
+ - **Model Type**: Image Captioning
16
+ - **Architecture**: CNN Encoder + LSTM Decoder
17
+ - **Framework**: PyTorch
18
+ - **Input**: Image (`.jpg`, `.png`, etc.)
19
+ - **Output**: Generated caption (text)
20
+ - **Vocabulary**: Pre-trained vocabulary file
21
+
22
+ ## 🚀 How to Use
23
+ ### **1️⃣ Install Dependencies**
24
+ ```bash
25
+ pip install torch torchvision transformers huggingface_hub pickle5