ShuhongLL / CEIR

Concept-based Explainable Image Representation

CEIR: Concept-based Explainable Image Representation

Demo

The trained models can be downloaded here. Please put the pretrained model to the directory "./saved_models_gpt4/[dataset_name]" and run the notebook in ./demo

Setup

1. Install Python (3.9) and PyTorch (2.0.1).
2. Install dependencies by running pip install -r requirements.txt

Reference

Label-free-CBM