Update README.md

README.md

@@ -4,22 +4,20 @@ tags:
 library_name: wildlife-datasets
 license: cc-by-nc-4.0
 ---
-# Model card for vit_small_patch14_dinov2.lvd142m
+# Model card for MegaDescriptor-B-224
 
-A Swin-V image feature model. Superwisely pre-trained on animal re-identification datasets.
+A Swin-B image feature model. Superwisely pre-trained on animal re-identification datasets.
 
 
 ## Model Details
 - **Model Type:** Animal re-identification / feature backbone
 - **Model Stats:**
   - Params (M): ??
-  - GMACs: ??
-  - Activations (M): ??
   - Image size: 224 x 224
 - **Papers:**
   - Swin Transformer: Hierarchical Vision Transformer using Shifted Windows --> https://arxiv.org/abs/2103.14030
 - **Original:** ??
-- **Pretrain Dataset:** ??
+- **Pretrain Dataset:** All available re-identification datasets --> TBD
 
 ## Model Usage
 ### Image Embeddings
@@ -47,16 +45,8 @@ output = model(train_transforms(img).unsqueeze(0))  # output is (batch_size, num
 # output is a (1, num_features) shaped tensor
 ```
 
-## Model Comparison
-???
-
 ## Citation
 
 ```bibtex
-@article{?????,
-  title={?????},
-  author={????},
-  journal={????},
-  year={????}
-}
+TBD
 ```