Skip to content

Latest commit

 

History

History
35 lines (33 loc) · 4.56 KB

README.md

File metadata and controls

35 lines (33 loc) · 4.56 KB
Raw

Visual Contrastive Learning for Few-shot Image Classification

This repository provides the code to classify images in two different categories, i.e. Similar (1) and Dissimilar (0) based on the image similarity task performed by utilizing a Contrastive Learning-based approach (including employing a custom contrastive loss). Furthermore, Siamese Networks is being used in n-way k-shot settings considered in the current implementation.

Requirements

  • Python 3.9
  • PyTorch 1.10.2
  • TorchVision 0.11.3
  • numpy 1.22.3
  • matplotlib 3.5.1

Usage

Data

Omniglot dataset is being used which is a collection of 1623 hand drawn characters from 50 different alphabets. For every character there are just 20 examples, each drawn by a different person. Each image is a gray scale image of resolution 105x105. Please clone this repo and then extract the images_background and images_evaluation folders. Finally, run DataGeneration.py file to create pickle files train.pickle and val.pickle files and store them in data folder. Here, train.pickle file contains characters from 30 different alphabets, whereas val.pickle contains characters from remaining 20 different alphabets.

Model Building and Training

  • The SiameseNetwork model class for n-way k-shot learning can be found in Model.py file.
  • To train the network, run Training.py file.
  • The average loss for the trained model is printed after every epoch.
  • All hyperparameters to control training and testing of the model are provided in the given Training.py file.

Output Samples

Image Similarity Scores

Image Comparison 1 Image Comparison 2 Image Comparion 3
alt text alt text alt text
Image Comparison 4 Image Comparison 5 Image Comparion 6
alt text alt text alt text
Image Comparison 7 Image Comparison 8 Image Comparion 9 Image Comparison 10
alt text alt text alt text alt text

Results for Image Classification

Image Comparison 1 Image Comparison 2 Image Comparion 3
alt text alt text alt text

Analysis

Among all the 10 comparisons made under Image Similarity Scores sub-section, images 1, 6, and 8 appear more similar, thereby having predicted labels as 1, as shown in the Results for Image Classification sub-section. This way, the current implementation frames the image similarity task as the image classification task.