Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,79 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: pytorch
|
| 3 |
+
tags:
|
| 4 |
+
- vision
|
| 5 |
+
- image-classification
|
| 6 |
+
- pytorch
|
| 7 |
+
license: mit
|
| 8 |
+
base_model: timm/maxvit_tiny_rw_224.sw_in1k
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# BTSbot
|
| 12 |
+
|
| 13 |
+
This is a maxvit fine-tuned for classifying images from the
|
| 14 |
+
Zwicky Transient Facility (ZTF) observatory.
|
| 15 |
+
[Rehemtulla et al. 2024](https://arxiv.org/abs/2401.15167) originally introduced
|
| 16 |
+
`BTSbot` and its classification task, and
|
| 17 |
+
[Rehemtulla et al. 2025](https://arxiv.org/abs/2512.11957) performed
|
| 18 |
+
architecture and pre-training benchmarking on this `BTSbot` image classification task.
|
| 19 |
+
|
| 20 |
+
**Base Model**:
|
| 21 |
+
[timm/maxvit_tiny_rw_224.sw_in1k](https://huggingface.co/timm/maxvit_tiny_rw_224.sw_in1k)
|
| 22 |
+
|
| 23 |
+
## Usage
|
| 24 |
+
|
| 25 |
+
Easily install the btsbot package and load this model with:
|
| 26 |
+
|
| 27 |
+
```python
|
| 28 |
+
git clone https://github.com/nabeelre/BTSbot.git
|
| 29 |
+
cd BTSbot
|
| 30 |
+
pip install -e .
|
| 31 |
+
|
| 32 |
+
import btsbot
|
| 33 |
+
model = btsbot.load_HF_model(
|
| 34 |
+
architecture="maxvit", multi_modal=False, pretrain="randinit"
|
| 35 |
+
)
|
| 36 |
+
```
|
| 37 |
+
|
| 38 |
+
Also see
|
| 39 |
+
[`BTSbot/btsbot/inference_example.py`](https://github.com/nabeelre/BTSbot/blob/main/btsbot/inference_example.py).
|
| 40 |
+
|
| 41 |
+
## Citation
|
| 42 |
+
|
| 43 |
+
If you use this model, please cite:
|
| 44 |
+
|
| 45 |
+
```bibtex
|
| 46 |
+
@ARTICLE{Rehemtulla+2025,
|
| 47 |
+
author = {{Rehemtulla}, Nabeel and {Miller}, Adam A. and {Walmsley}, Mike
|
| 48 |
+
and {Shah}, Ved G. and {Jegou du Laz}, Theophile and
|
| 49 |
+
{Coughlin}, Michael W. and {Sasli}, Argyro and
|
| 50 |
+
{Bloom}, Joshua and {Fremling}, Christoffer and
|
| 51 |
+
{Graham}, Matthew J. and {Groom}, Steven L. and {Hale}, David and
|
| 52 |
+
{Mahabal}, Ashish A. and {Perley}, Daniel A. and
|
| 53 |
+
{Purdum}, Josiah and {Rusholme}, Ben and {Sollerman}, Jesper and
|
| 54 |
+
{Kasliwal}, Mansi M.},
|
| 55 |
+
title = "{Pre-training vision models for the classification of alerts from
|
| 56 |
+
wide-field time-domain surveys}",
|
| 57 |
+
journal = {arXiv e-prints},
|
| 58 |
+
keywords = {Instrumentation and Methods for Astrophysics,
|
| 59 |
+
Computer Vision and Pattern Recognition},
|
| 60 |
+
year = 2025,
|
| 61 |
+
month = dec,
|
| 62 |
+
eid = {arXiv:2512.11957},
|
| 63 |
+
pages = {arXiv:2512.11957},
|
| 64 |
+
doi = {10.48550/arXiv.2512.11957},
|
| 65 |
+
archivePrefix = {arXiv},
|
| 66 |
+
eprint = {2512.11957},
|
| 67 |
+
primaryClass = {astro-ph.IM},
|
| 68 |
+
adsurl = {https://ui.adsabs.harvard.edu/abs/2025arXiv251211957R},
|
| 69 |
+
adsnote = {Provided by the SAO/NASA Astrophysics Data System}
|
| 70 |
+
}
|
| 71 |
+
```
|
| 72 |
+
|
| 73 |
+
## License
|
| 74 |
+
|
| 75 |
+
This model is released under the MIT License.
|
| 76 |
+
|
| 77 |
+
## Repository
|
| 78 |
+
|
| 79 |
+
For more information, see the [BTSbot GitHub repository](https://github.com/nabeelre/BTSbot).
|