Update README.md
Browse files
README.md
CHANGED
|
@@ -9,7 +9,7 @@ base_model:
|
|
| 9 |
- facebook/PE-Core-B16-224
|
| 10 |
---
|
| 11 |
|
| 12 |
-
# Model Card for
|
| 13 |
|
| 14 |
A ViT-B16 image encoder from the PE-Core model by Bolya et al., converted to the Birder format for image feature extraction.
|
| 15 |
This version retains the original model weights and architecture, with the exception of removing the CLIP projection layer to expose raw image embeddings.
|
|
@@ -37,7 +37,7 @@ See: <https://huggingface.co/facebook/PE-Core-B16-224> for further details.
|
|
| 37 |
import birder
|
| 38 |
from birder.inference.classification import infer_image
|
| 39 |
|
| 40 |
-
(net, model_info) = birder.load_pretrained_model("
|
| 41 |
|
| 42 |
# Get the image size the model was trained on
|
| 43 |
size = birder.get_size_from_signature(model_info.signature)
|
|
@@ -56,7 +56,7 @@ image = "path/to/image.jpeg" # or a PIL image
|
|
| 56 |
from PIL import Image
|
| 57 |
import birder
|
| 58 |
|
| 59 |
-
(net, model_info) = birder.load_pretrained_model("
|
| 60 |
|
| 61 |
# Get the image size the model was trained on
|
| 62 |
size = birder.get_size_from_signature(model_info.signature)
|
|
|
|
| 9 |
- facebook/PE-Core-B16-224
|
| 10 |
---
|
| 11 |
|
| 12 |
+
# Model Card for rope_i_vit_b16_pn_aps_c1_pe-core
|
| 13 |
|
| 14 |
A ViT-B16 image encoder from the PE-Core model by Bolya et al., converted to the Birder format for image feature extraction.
|
| 15 |
This version retains the original model weights and architecture, with the exception of removing the CLIP projection layer to expose raw image embeddings.
|
|
|
|
| 37 |
import birder
|
| 38 |
from birder.inference.classification import infer_image
|
| 39 |
|
| 40 |
+
(net, model_info) = birder.load_pretrained_model("rope_i_vit_b16_pn_aps_c1_pe-core", inference=True)
|
| 41 |
|
| 42 |
# Get the image size the model was trained on
|
| 43 |
size = birder.get_size_from_signature(model_info.signature)
|
|
|
|
| 56 |
from PIL import Image
|
| 57 |
import birder
|
| 58 |
|
| 59 |
+
(net, model_info) = birder.load_pretrained_model("rope_i_vit_b16_pn_aps_c1_pe-core", inference=True)
|
| 60 |
|
| 61 |
# Get the image size the model was trained on
|
| 62 |
size = birder.get_size_from_signature(model_info.signature)
|