Merge

Merge branch 'main' of https://github.com/ssundaram21/dreamsim

README.md

@@ -66,6 +66,14 @@ distance = model(img1, img2) # The model takes an RGB image from [0, 1], size ba
 
 To run on example images, run `demo.py`. The script should produce distances (0.424, 0.34). 
 
+### (new!) Single-branch models
+By default, DreamSim uses an ensemble of CLIP, DINO, and OpenCLIP (all ViT-B/16). If you need a lighter-weight model you can use *single-branch* versions of DreamSim where only a single backbone is finetuned. The available options are OpenCLIP-ViTB/32, DINO-ViTB/16, CLIP-ViTB/32, in order of performance. 
+
+To load a single-branch model, use the `dreamsim_type` argument. For example:
+```
+dreamsim_dino_model, preprocess = dreamsim(pretrained=True, dreamsim_type="dino_vitb16")
+```
+
 ### Feature extraction
 To extract a *single image embedding* using dreamsim, use the `embed` method as shown in the following snippet:
 ```