Update README.md
Browse files
README.md
CHANGED
|
@@ -8,7 +8,7 @@ license: apache-2.0
|
|
| 8 |
|
| 9 |
# Model Card for davit_tiny_il-all
|
| 10 |
|
| 11 |
-
A Dual Attention Vision Transformer (DaViT) image classification model. This model was trained on the `il-all` dataset
|
| 12 |
|
| 13 |
The species list is derived from data available at <https://www.israbirding.com/checklist/>.
|
| 14 |
|
|
@@ -16,12 +16,12 @@ The species list is derived from data available at <https://www.israbirding.com/
|
|
| 16 |
|
| 17 |
- **Model Type:** Image classification and detection backbone
|
| 18 |
- **Model Stats:**
|
| 19 |
-
|
| 20 |
-
|
| 21 |
- **Dataset:** il-all (550 classes)
|
| 22 |
|
| 23 |
- **Papers:**
|
| 24 |
-
|
| 25 |
|
| 26 |
## Model Usage
|
| 27 |
|
|
@@ -39,9 +39,9 @@ size = birder.get_size_from_signature(signature)
|
|
| 39 |
# Create an inference transform
|
| 40 |
transform = birder.classification_transform(size, rgb_stats)
|
| 41 |
|
| 42 |
-
image = "path/to/image.jpeg" # or a PIL image
|
| 43 |
(out, _) = infer_image(net, image, transform)
|
| 44 |
-
# out is a NumPy array with shape of (1, num_classes)
|
| 45 |
```
|
| 46 |
|
| 47 |
### Image Embeddings
|
|
|
|
| 8 |
|
| 9 |
# Model Card for davit_tiny_il-all
|
| 10 |
|
| 11 |
+
A Dual Attention Vision Transformer (DaViT) image classification model. This model was trained on the `il-all` dataset, encompassing all relevant bird species found in Israel, including rarities.
|
| 12 |
|
| 13 |
The species list is derived from data available at <https://www.israbirding.com/checklist/>.
|
| 14 |
|
|
|
|
| 16 |
|
| 17 |
- **Model Type:** Image classification and detection backbone
|
| 18 |
- **Model Stats:**
|
| 19 |
+
- Params (M): 28.0
|
| 20 |
+
- Input image size: 384 x 384
|
| 21 |
- **Dataset:** il-all (550 classes)
|
| 22 |
|
| 23 |
- **Papers:**
|
| 24 |
+
- DaViT: Dual Attention Vision Transformers: <https://arxiv.org/abs/2204.03645>
|
| 25 |
|
| 26 |
## Model Usage
|
| 27 |
|
|
|
|
| 39 |
# Create an inference transform
|
| 40 |
transform = birder.classification_transform(size, rgb_stats)
|
| 41 |
|
| 42 |
+
image = "path/to/image.jpeg" # or a PIL image, must be loaded in RGB format
|
| 43 |
(out, _) = infer_image(net, image, transform)
|
| 44 |
+
# out is a NumPy array with shape of (1, num_classes), representing class probabilities.
|
| 45 |
```
|
| 46 |
|
| 47 |
### Image Embeddings
|