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