waterfall109 commited on
Commit
d3640c8
·
1 Parent(s): 749a146

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -83
README.md CHANGED
@@ -3,66 +3,42 @@ language:
3
  - en
4
  license: gpl-2.0
5
  tags:
6
- - radio-astronomy
7
- - fast-radio-bursts
8
- - pulsar
9
- - instance-segmentation
10
- - mask-rcnn
11
- - signal-processing
12
  library_name: pytorch
13
  pipeline_tag: image-segmentation
14
  datasets:
15
  - CRAFTS-FRT
16
  metrics:
17
  - recall
18
- - precision
19
  ---
20
- # FRTSearch: Fast Radio Transient Detection Model
21
 
22
- [![Paper](https://img.shields.io/badge/Paper-AASTeX-blue.svg)](https://doi.org/10.57760/sciencedb.Fastro.00038)
23
- [![Dataset](https://img.shields.io/badge/Dataset-CRAFTS--FRT-yellow.svg)](https://doi.org/10.57760/sciencedb.Fastro.00038)
24
- [![GitHub](https://img.shields.io/badge/GitHub-FRTSearch-black.svg)](https://github.com/BinZhang109/FRTSearch)
25
- [![Python](https://img.shields.io/badge/Python-3.10+-green.svg)](https://www.python.org/)
26
- [![License](https://img.shields.io/badge/License-GPL--2.0-grey.svg)](https://github.com/BinZhang109/FRTSearch/blob/main/LICENSE)
27
 
28
- ---
29
-
30
- ## Model Description
31
-
32
- **FRTSearch** is an end-to-end deep learning framework for detecting and characterizing Fast Radio Transients (FRTs), including:
33
- - **Fast Radio Bursts (FRBs)**
34
- - **Pulsars**
35
- - **Rotating Radio Transients (RRATs)**
36
-
37
- ### Architecture
38
-
39
- - **Backbone**: HRNet-W32 (High-Resolution Network)
40
- - **Detection Head**: Mask R-CNN
41
- - **Training Epoch**: 36
42
- - **Input Dimensions**: 256 (frequency channels) × 8192 (time samples)
43
- - **Model Size**: 381.75 MB
44
- ---
45
-
46
- ## Supported Data Formats
47
 
48
- - **PSRFITS** (`.fits`) - Standard format for pulsar data
49
- - **Sigproc Filterbank** (`.fil`) - Legacy filterbank format
50
 
51
- ### Supported Bit Depths
52
 
53
- - 1-bit quantization
54
- - 2-bit quantization
55
- - 4-bit quantization
56
- - 8-bit quantization
57
- - 32-bit quantization
58
- ---
 
59
 
60
- ## Download Model Weights
61
 
62
  ```python
63
  from huggingface_hub import hf_hub_download
64
 
65
- # Download model weights
66
  model_path = hf_hub_download(
67
  repo_id="waterfall109/FRTSearch",
68
  filename="models/hrnet_epoch_36.pth"
@@ -71,39 +47,23 @@ model_path = hf_hub_download(
71
 
72
  Or download directly from [Files and versions](https://huggingface.co/waterfall109/FRTSearch/tree/main).
73
 
74
- ---
75
 
76
  ## Test Samples
77
 
78
  This repository includes 7 test samples from 3 different telescopes to demonstrate cross-facility performance:
79
 
80
- ### FAST Telescope (3 samples)
81
- - FRB 20121102 (DM=565.0 pc cm⁻³)
82
- - FRB 20180301 (DM=420.0 pc cm⁻³)
83
- - FRB 20201124 (DM=525.0 pc cm⁻³)
84
-
85
- ### SKA/ASKAP (3 samples)
86
- - FRB 20180119 (DM=400.0 pc cm⁻³)
87
- - FRB 20180212 (DM=167.7 pc cm⁻³)
88
- - FRB 20220610A (DM=1457.6 pc cm⁻³)
89
-
90
- ### Parkes Telescope (1 sample)
91
- - FRB 110220 (DM=944.0 pc cm⁻³)
92
-
93
- See [DATA_DESCRIPTION.md](DATA_DESCRIPTION.md) for detailed information about each test sample.
94
-
95
- ---
96
-
97
 
98
  ## Citation
99
 
100
- If you use FRTSearch or the CRAFTS-FRT dataset in your research, please cite:
101
-
102
  ```bibtex
103
  @article{zhang2025frtsearch,
104
- title={FRTSearch: Unified Detection and Parameter Inference of Fast Radio Transients using Instance Segmentation},
105
  author={Zhang, Bin and Wang, Yabiao and Xie, Xiaoyao and others},
106
- journal={Draft version (AASTeX631)},
107
  year={2025}
108
  }
109
  ```
@@ -116,26 +76,14 @@ When using the test samples, please also cite the original observations:
116
  - **Parkes sample**: [Keane et al. (2015)](https://doi.org/10.1093/mnras/stu2650)
117
  - **SKA samples**: [Shannon et al. (2018)](https://doi.org/10.1038/s41586-018-0588-y), [Ryder et al. (2022)](https://doi.org/10.1126/science.adf2678)
118
 
119
- ---
120
-
121
- ## License
122
 
123
- This project is based on [PRESTO](https://github.com/scottransom/presto) and modified components are licensed under GNU General Public License v2.0.
124
-
125
- See [LICENSE](https://github.com/BinZhang109/FRTSearch/blob/main/LICENSE) for details.
126
 
127
  ---
128
-
129
- ## Acknowledgments
130
-
131
- - **Dataset**: CRAFTS-FRT from FAST telescope observations
132
- - **Framework**: Built on [MMDetection](https://github.com/open-mmlab/mmdetection)
133
- - **Base Code**: Modified from [PRESTO](https://github.com/scottransom/presto)
134
-
135
- ---
136
-
137
  <div align="center">
138
- <sub>Exploring the dynamic universe with AI 🌌📡</sub>
139
- <br>
140
- <sub>For questions and issues: <a href="https://github.com/BinZhang109/FRTSearch/issues">GitHub Issues</a></sub>
141
  </div>
 
 
 
 
3
  - en
4
  license: gpl-2.0
5
  tags:
6
+ - Radio Astronomy
7
+ - Pulsar
8
+ - Rotating Radio Transients
9
+ - Fast Radio Bursts
10
+ - Mask-RCNN
11
+ - Signal Processing
12
  library_name: pytorch
13
  pipeline_tag: image-segmentation
14
  datasets:
15
  - CRAFTS-FRT
16
  metrics:
17
  - recall
18
+ - FPR
19
  ---
 
20
 
21
+ # FRTSearch: Fast Radio Transient Detection
 
 
 
 
22
 
23
+ [![Paper](https://img.shields.io/badge/Paper-AASTeX-blue.svg)](https://doi.org/10.57760/sciencedb.Fastro.00038) [![Dataset](https://img.shields.io/badge/Dataset-CRAFTS--FRT-yellow.svg)](https://doi.org/10.57760/sciencedb.Fastro.00038) [![GitHub](https://img.shields.io/badge/GitHub-FRTSearch-black.svg)](https://github.com/BinZhang109/FRTSearch)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
24
 
25
+ **FRTSearch** is an end-to-end deep learning framework for detecting and characterizing Fast Radio Transients (FRTs), including: **Pulsars**, and **Rotating Radio Transients (RRATs)** **Fast Radio Bursts (FRBs)**.
 
26
 
27
+ ## Model Info
28
 
29
+ | Item | Value |
30
+ |------|-------|
31
+ | Backbone | HRNet-W32 |
32
+ | Input | 256 × 8192 (freq × time) |
33
+ | Size | 381.75 MB |
34
+ | Formats | `.fits` (PSRFITS), `.fil` (Filterbank) |
35
+ | Bit Depth | 1/2/4/8/32-bit |
36
 
37
+ ## Quick Start
38
 
39
  ```python
40
  from huggingface_hub import hf_hub_download
41
 
 
42
  model_path = hf_hub_download(
43
  repo_id="waterfall109/FRTSearch",
44
  filename="models/hrnet_epoch_36.pth"
 
47
 
48
  Or download directly from [Files and versions](https://huggingface.co/waterfall109/FRTSearch/tree/main).
49
 
 
50
 
51
  ## Test Samples
52
 
53
  This repository includes 7 test samples from 3 different telescopes to demonstrate cross-facility performance:
54
 
55
+ | Telescope | FRB | DM (pc cm⁻³) |
56
+ |-----------|-----|--------------|
57
+ | FAST | 20121102, 20180301, 20201124 | 565, 420, 525 |
58
+ | ASKAP | 20180119, 20180212, 20220610A | 400, 168, 1458 |
59
+ | Parkes | 110220 | 944 |
 
 
 
 
 
 
 
 
 
 
 
 
60
 
61
  ## Citation
62
 
 
 
63
  ```bibtex
64
  @article{zhang2025frtsearch,
65
+ title={FRTSearch: Unified Detection and Parameter Inference of Fast Radio Transients},
66
  author={Zhang, Bin and Wang, Yabiao and Xie, Xiaoyao and others},
 
67
  year={2025}
68
  }
69
  ```
 
76
  - **Parkes sample**: [Keane et al. (2015)](https://doi.org/10.1093/mnras/stu2650)
77
  - **SKA samples**: [Shannon et al. (2018)](https://doi.org/10.1038/s41586-018-0588-y), [Ryder et al. (2022)](https://doi.org/10.1126/science.adf2678)
78
 
79
+ ## License & Acknowledgments
 
 
80
 
81
+ GPL-2.0 | Based on [MMDetection](https://github.com/open-mmlab/mmdetection) & [PRESTO](https://github.com/scottransom/presto)
 
 
82
 
83
  ---
 
 
 
 
 
 
 
 
 
84
  <div align="center">
85
+ <sub>🌌 Questions? <a href="https://github.com/BinZhang109/FRTSearch/issues">GitHub Issues</a></sub>
 
 
86
  </div>
87
+ ```
88
+
89
+ ---