sat2cap added

mvrl · Aug 21, 2024 · 981d1c0 · 981d1c0
1 parent 736b446
commit 981d1c0
Show file tree

Hide file tree

Showing 6 changed files with 37 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -26,9 +26,11 @@ print(model.forward_encoder(torch.randn(1, 3, 224, 224), mask_ratio=0.0)[0].shap
 |Prithvi||[link](./rshf/prithvi/README.md)
 |RemoteCLIP|TGRS'23|[link](./rshf/remoteclip/README.md)
 |RVSA|TGRS'22|[link](./rshf/rvsa/README.md)
+|Sat2Cap|EarthVision'24|[link](./rshf/sat2cap/README.md)
 |SatClip||[link](./rshf/satclip/README.md)
 |SatMAE|NeurIPS'22|[link](./rshf/satmae/README.md)
 |SatMAE++|CVPR'24|[link](./rshf/satmaepp/README.md)
 |ScaleMAE|ICCV'23|[link](./rshf/scalemae/README.md)
 |SINR|ICML'23|[link](./rshf/sinr/README.md)
+|StreetCLIP||[link](./rshf/streetclip/README.md)
 ### List of models available here: [Link](https://huggingface.co/collections/MVRL/remote-sensing-foundation-models-664e8fcd67d8ca8c03f42d00)
diff --git a/rshf/sat2cap/README.md b/rshf/sat2cap/README.md
@@ -0,0 +1,16 @@
+## Sat2Cap
+
+References
+
+Arxiv: https://arxiv.org/abs/2307.15904 \
+GitHub: https://github.com/mvrl/Sat2Cap \
+Citation:
+```bib
+@inproceedings{dhakal2024sat2cap,
+  title={Sat2cap: Mapping fine-grained textual descriptions from satellite images},
+  author={Dhakal, Aayush and Ahmad, Adeel and Khanal, Subash and Sastry, Srikumar and Kerner, Hannah and Jacobs, Nathan},
+  booktitle={IEEE/ISPRS Workshop: Large Scale Computer Vision for Remote Sensing (EARTHVISION)},
+  pages={533--542},
+  year={2024}
+}
+```
diff --git a/rshf/sat2cap/__init__.py b/rshf/sat2cap/__init__.py
@@ -0,0 +1 @@
+from transformers import CLIPVisionModel as Sat2Cap
diff --git a/rshf/streetclip/README.md b/rshf/streetclip/README.md
@@ -0,0 +1,16 @@
+## StreetCLIP
+
+References
+
+Arxiv: https://arxiv.org/abs/2302.00275 \
+Citation:
+```bib
+  @misc{haas2023learning,
+      title={Learning Generalized Zero-Shot Learners for Open-Domain Image Geolocalization}, 
+      author={Lukas Haas and Silas Alberti and Michal Skreta},
+      year={2023},
+      eprint={2302.00275},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV}
+  }
+```
diff --git a/rshf/streetclip/__init__.py b/rshf/streetclip/__init__.py
@@ -0,0 +1 @@
+from transformers import CLIPModel as StreetCLIP
diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = rshf
-version = 0.0.10
+version = 0.0.11
 author = Srikumar Sastry
 author_email = s.sastry@wustl.edu
 description = RS pretrained models in huggingface style
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		from transformers import CLIPVisionModel as Sat2Cap
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		from transformers import CLIPModel as StreetCLIP