Simulation Dataset for BBHnet

NOTE: OLD README, UPDATE IN PROGRESS

We generate simulation dataset to train BBHnet, our deep learning framework for detection of compact binary coalescene (CBC) gravitational-wave (GW) signals .

Example

To generate a noise dataset, simply run generateRealNoise.py:

python generateRealNoise.py -t0 1186729980 -t1 1186734086 -t0-psd 1186729980 -t1-psd 1186734086
    -fs 1024 -fmin 20 -o test_noise.h5

To also add CBC signals, enable the flag -S and add the prior distribution file in Bilby format with -p

python generateRealNoise.py -t0 1186729980 -t1 1186734086 -t0-psd 1186729980 -t1-psd 1186734086
    -fs 1024 -fmin 20 -S -p config/priors/nonspin_BBH.prior -o test_signal.h5

A full list of generateRealNoise.py arguments can be found below:

usage: generateRealNoise.py [-h] -t0 FRAME_START -t1 FRAME_STOP -t0-psd FRAME_START_PSD -t1-psd FRAME_STOP_PSD -o OUTFILE [-S]
                            [-fs SAMPLE_RATE] [-fmin HIGH_PASS] [-T SAMPLE_DURATION] [-dt TIME_STEP] [-p PRIOR_FILE]
                            [--correlation-shift CORRELATION_SHIFT] [--min-trigger MIN_TRIGGER] [--max-trigger MAX_TRIGGER]
                            [-s SEED]

optional arguments:
  -h, --help            show this help message and exit
  -t0 FRAME_START, --frame-start FRAME_START
                        starting GPS time of strain
  -t1 FRAME_STOP, --frame-stop FRAME_STOP
                        stopping GPS time of strain
  -t0-psd FRAME_START_PSD, --frame-start-psd FRAME_START_PSD
                        starting GPS time of strain for PSD estimation
  -t1-psd FRAME_STOP_PSD, --frame-stop-psd FRAME_STOP_PSD
                        stopping GPS time of strain for PSD estimation
  -o OUTFILE, --outfile OUTFILE
                        path to write output file in HDF5 format
  -S, --signal          Enable to add GW signal on top of background noise
  -fs SAMPLE_RATE, --sample-rate SAMPLE_RATE
                        sampling rate of strain
  -fmin HIGH_PASS, --high-pass HIGH_PASS
                        frequency of highpass filter
  -T SAMPLE_DURATION, --sample-duration SAMPLE_DURATION
                        duration in seconds of each sample
  -dt TIME_STEP, --time-step TIME_STEP
                        time step size in seconds between consecutive samples
  -p PRIOR_FILE, --prior-file PRIOR_FILE
                        path to prior config file. Required for signal simulation
  --correlation-shift CORRELATION_SHIFT
                        if given, also compute the correlation with given shift value
  --min-trigger MIN_TRIGGER
                        mininum trigger time w.r.t to sample. must be within [0, sample_duration]
  --max-trigger MAX_TRIGGER
                        maximum trigger time w.r.t to sample. must be within [0, sample_duration]
  -s SEED, --seed SEED  random seed for reproducibility

Simulation code and tutorial for BBHnet training data

Related tags

Overview

Simulation Dataset for BBHnet

NOTE: OLD README, UPDATE IN PROGRESS

Example

Owner

PFENet: Prior Guided Feature Enrichment Network for Few-shot Segmentation (TPAMI).

Current state of supervised and unsupervised depth completion methods

NAACL'2021: Factual Probing Is [MASK]: Learning vs. Learning to Recall

Gems & Holiday Package Prediction

OpenDILab Multi-Agent Environment

The repository offers the official implementation of our paper in PyTorch.

Reverse engineering recurrent neural networks with Jacobian switching linear dynamical systems

Coded illumination for improved lensless imaging

A simple algorithm for extracting tree height in sparse scene from point cloud data.

Generative Flow Networks for Discrete Probabilistic Modeling

An open source Python package for plasma science that is under development

Implements a fake news detection program using classifiers.

This is the official code release for the paper Shape and Material Capture at Home

Breaking the Dilemma of Medical Image-to-image Translation

Mixed Neural Likelihood Estimation for models of decision-making

Image Segmentation using U-Net, U-Net with skip connections and M-Net architectures

Weakly Supervised Segmentation by Tensorflow.

Resources for the Ki testnet challenge

Learning infinite-resolution image processing with GAN and RL from unpaired image datasets, using a differentiable photo editing model.

Implementation of Ag-Grid component for Streamlit