File size: 3,170 Bytes
4457836 4c36442 a229cde 4c36442 4457836 a229cde 4c36442 a229cde 4c36442 a229cde 4c36442 a229cde 4c36442 a229cde 4c36442 a229cde b15fc35 a229cde 4c36442 fc3d3e7 4c36442 b15fc35 fc3d3e7 4c36442 b15fc35 fc3d3e7 4c36442 fc3d3e7 80bb7a8 fc3d3e7 4c36442 fc3d3e7 4c36442 | 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 | ---
license: mit
library_name: pytorch
tags:
- astronomy
- fast-radio-burst
- transient-search
- object-detection
- image-classification
---
# DRAFTS Model Repository
Pretrained model weights for
[DRAFTS](https://github.com/SukiYume/DRAFTS), a deep-learning radio fast
transient search pipeline.
## Current search runtime
These are the two checkpoints used by the current DRAFTS observation-search
and injection-search entry points:
| Task | Architecture | File | Size | SHA-256 |
|---|---|---|---:|---|
| Candidate localization | CenterNet + ConvNeXt-Tiny, v10 | `object_best_model_centernet_conv_tiny_ema_v10.pth` | 114,449,795 bytes | `bcad4e710f5f1ccd3c8609d35a8d3fbfc36abd1d85bfefd035e945a573fb0629` |
| Candidate verification | ConvNeXt-Small binary classifier | `binary_best_model_conv_small_ema.pth` | 197,949,537 bytes | `2055745aab76ddc16074516aa7b9aafdfaedf16df37ce8924389573eab27ffd8` |
Download both files with the Hugging Face CLI:
```bash
mkdir -p runcode/models
hf download TorchLight/DRAFTS \
object_best_model_centernet_conv_tiny_ema_v10.pth \
binary_best_model_conv_small_ema.pth \
--local-dir runcode/models
```
Or use direct HTTP downloads:
```bash
curl -L \
-o runcode/models/object_best_model_centernet_conv_tiny_ema_v10.pth \
https://huggingface.co/TorchLight/DRAFTS/resolve/main/object_best_model_centernet_conv_tiny_ema_v10.pth
curl -L \
-o runcode/models/binary_best_model_conv_small_ema.pth \
https://huggingface.co/TorchLight/DRAFTS/resolve/main/binary_best_model_conv_small_ema.pth
```
The injection runtime uses the same filenames under
`injection_experiment/search_runtime/models/`. Compare the downloaded files
with the SHA-256 values above before running a long search.
Model architecture definitions, preprocessing, command-line examples, and
deployment notes are maintained in the DRAFTS repository:
- [Observation-search runtime](https://github.com/SukiYume/DRAFTS/tree/main/runcode#readme)
- [CenterNet detector training](https://github.com/SukiYume/DRAFTS/tree/main/object_detection#readme)
- [Binary-classifier training](https://github.com/SukiYume/DRAFTS/tree/main/binary_classification#readme)
- [Injection experiment](https://github.com/SukiYume/DRAFTS/tree/main/injection_experiment#readme)
## Legacy checkpoints
Earlier ResNet detector and classifier checkpoints are grouped under
[`legacy/`](legacy/). They remain available for reproducing older DRAFTS
experiments, but they are not used by the current runtime.
| Task | File |
|---|---|
| CenterNet ResNet-18 detector | [`legacy/cent_resnet18.pth`](legacy/cent_resnet18.pth) |
| CenterNet ResNet-50 detector | [`legacy/cent_resnet50.pth`](legacy/cent_resnet50.pth) |
| ResNet-18 binary classifier | [`legacy/class_resnet18.pth`](legacy/class_resnet18.pth) |
| ResNet-50 binary classifier | [`legacy/class_resnet50.pth`](legacy/class_resnet50.pth) |
## License and citation
The accompanying DRAFTS code is released under the MIT License. If DRAFTS is
useful in your work, cite
[DRAFTS: A Deep Learning-Based Radio Fast Transient Search Pipeline](https://arxiv.org/abs/2410.03200)
and record the exact checkpoint and search configuration used.
|