FRTSearch is an end-to-end framework for discovering Pulsars, Rotating Radio Transients (RRATs), and Fast Radio Bursts (FRBs) in radio astronomical observation data. Single-pulse emissions from these sources all exhibit consistent dispersive trajectories governed by the cold plasma dispersion relation ($t \propto \nu^{-2}$) in time-frequency dynamic spectra. This shared signature serves as a key beacon for identifying these astrophysical sources. FRTSearch leverages a Mask R-CNN instance segmentation model and the IMPIC algorithm to directly detect and characterize Fast Radio Transients (FRTs), infer their physical parameters (DM, ToA), and generate diagnostic plots and candidate catalogs for manual verification and scientific analysis.

Core Components:

1. Mask R-CNN — Segments dispersive trajectories in time-frequency dynamic spectra, trained on the pixel-level annotated CRAFTS-FRT dataset.
2. IMPIC — Iterative Mask-based Parameter Inference and Calibration: infers DM and ToA directly from segmentation masks.   Code | Docs | Example

Supported formats:

Requires: Python 3.10+, CUDA 11.7+, PyTorch 2.0+, PRESTO, MMDetection

pip install FRTSearch

git clone https://github.com/BinZhang109/FRTSearch.git && cd FRTSearch
pip install -r requirements.txt

docker pull binzhang109/frtsearch:v1.0.0

Download from Hugging Face and place into models/:

FRTSearch/
├── models/
│   └── hrnet_epoch_36.pth
├── configs/
│   ├── detector_FAST.py
│   └── detector_SKA.py
└── ...

python FRTSearch.py <data.fits|data.fil> <config.py> [--slide-size 128]

Test data can be downloaded from Hugging Face.

# FAST FRB detection
python FRTSearch.py ./test_sample/FRB20121102_0038.fits ./configs/detector_FAST.py --slide-size 128

# SKA FRB detection
python FRTSearch.py ./test_sample/FRB20180119_SKA_1660_1710.fil ./configs/detector_SKA.py --slide-size 8

Download the CRAFTS-FRT dataset and place it into CRAFTS_FRT_Dataset/ before training.

python train.py

python test_sample/test_samples.py --example FRB20121102

Available examples: FRB20121102, FRB20201124, FRB20180301, FRB20180119, FRB20180212

The first pixel-level annotated FRT dataset, derived from the Commensal Radio Astronomy FAST Survey (CRAFTS).

@article{zhang2026frtsearch,
  title={FRTSearch: Unified Detection and Parameter Inference of Fast Radio Transients using Instance Segmentation },
  author={Zhang, Bin and Wang, Yabiao and Xie, Xiaoyao et al.}
  year={2026},
}

Test sample references: FAST — Guo et al. (2025)  |  SKA — Shannon et al. (2018)

Open an Issue for bugs or questions. PRs welcome — see Contributing Guidelines.

This project is licensed under GPL-2.0.

Built upon: MMDetection | PRESTO