Image Captioning Model Comparison

Overview

As a member of an Agile team of six, I contributed to the development and comparison of four lightweight encoder-decoder models for image captioning using Keras. We employed a modified pretrained VGG16 convolutional neural network to extract image features from the Flickr8k dataset. Throughout this process, I gained significant knowledge and skills in deep learning, model optimization, and teamwork. To showcase our work, I created an interactive website with a React.js frontend and Flask backend, demonstrating the models and their bidirectional variants at the Canadian Undergraduate Conference on AI (CUCAI). Additionally, I co-authored and published a research paper for CUCAI, presenting an in-depth analysis of our experiment's findings. This project was a tremendous learning experience, enhancing my technical expertise and collaborative abilities.