Document custom zero-shot labels
Browse files
README.md
CHANGED
|
@@ -121,6 +121,15 @@ for label, score in classifier.predict(image, top_k=5):
|
|
| 121 |
print(label, score)
|
| 122 |
```
|
| 123 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 124 |
## Evaluate Zero-Shot Accuracy
|
| 125 |
|
| 126 |
The dataset should be arranged like `torchvision.datasets.ImageFolder`:
|
|
|
|
| 121 |
print(label, score)
|
| 122 |
```
|
| 123 |
|
| 124 |
+
Use custom candidate labels by passing them to `predict`:
|
| 125 |
+
|
| 126 |
+
```python
|
| 127 |
+
custom_labels = ["cacao", "coffee", "mango", "olive", "sunflower"]
|
| 128 |
+
|
| 129 |
+
for label, score in classifier.predict(image, labels=custom_labels, top_k=5):
|
| 130 |
+
print(label, score)
|
| 131 |
+
```
|
| 132 |
+
|
| 133 |
## Evaluate Zero-Shot Accuracy
|
| 134 |
|
| 135 |
The dataset should be arranged like `torchvision.datasets.ImageFolder`:
|