Skip to content

Latest commit

 

History

History

clip-zero-shot-image-classification

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

Zero-shot Image Classification with OpenAI CLIP

Zero-shot image classification is a computer vision task to classify images into one of several classes, without any prior training or knowledge of the classes.

zero-shot-pipeline

In this tutorial, you will use OpenAI CLIP model to perform zero-shot image classification.

Notebook Contents

This tutorial demonstrates how to perform zero-shot image classification using the open-source CLIP model. CLIP is a multi-modal vision and language model. It can be instructed in natural language to predict the most relevant text snippet, given an image, without directly optimizing for the task. According to the paper, CLIP matches the performance of the original ResNet50 on ImageNet “zero-shot” without using any of the original 1.28M labeled examples, overcoming several major challenges in computer vision. You can find more information about this model in the research paper, OpenAI blog, model card and GitHub repository.

This folder contains notebook that show how to convert and quantize model with OpenVINO and NNCF

The notebook contains the following steps:

  1. Download the model.
  2. Instantiate the PyTorch model.
  3. Convert model to OpenVINO IR, using the model conversion API.
  4. Run CLIP with OpenVINO.
  5. Quantize the converted model with NNCF.
  6. Check the quantized model inference result.
  7. Compare model size of converted and quantized models.
  8. Compare performance of converted and quantized models.
  9. Launch interactive demo

We will use CLIP model for zero-shot image classification. The result of model work demonstrated on the image below image

Installation Instructions

This is a self-contained example that relies solely on its own code.
We recommend running the notebook in a virtual environment. You only need a Jupyter server to start. For details, please refer to Installation Guide..