nabeelr commited on
Commit
e478ed1
·
verified ·
1 Parent(s): 6bafd25

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +79 -0
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).