This is a fixed-text formatted version of a Jupyter notebook
You can contribute with your own notebooks in this GitHub repository.
Source files: detect.ipynb | detect.py
Source detection with Gammapy¶
Context¶
The first task in a source catalogue production is to identify significant excesses in the data that can be associated to unknown sources and provide a preliminary parametrization in term of position, extent, and flux. In this notebook we will use Fermi-LAT data to illustrate how to detect candidate sources in counts images with known background.
Objective: build a list of significant excesses in a Fermi-LAT map
Proposed approach¶
This notebook show how to do source detection with Gammapy using the methods available in gammapy.estimators
. We will use images from a Fermi-LAT 3FHL high-energy Galactic center dataset to do this:
perform adaptive smoothing on counts image
produce 2-dimensional test-statistics (TS)
run a peak finder to detect point-source candidates
compute Li & Ma significance images
estimate source candidates radius and excess counts
Note that what we do here is a quick-look analysis, the production of real source catalogs use more elaborate procedures.
We will work with the following functions and classes:
gammapy.estimators.utils.find_peaks
Setup¶
As always, let’s get started with some setup …
[1]:
%matplotlib inline
import matplotlib.pyplot as plt
import matplotlib.cm as cm
from gammapy.maps import Map
from gammapy.estimators import ASmoothMapEstimator, TSMapEstimator
from gammapy.estimators.utils import find_peaks
from gammapy.datasets import MapDataset
from gammapy.modeling.models import (
BackgroundModel,
SkyModel,
PowerLawSpectralModel,
PointSpatialModel,
)
from gammapy.irf import PSFMap, EnergyDependentTablePSF, EDispKernelMap
from astropy.coordinates import SkyCoord
import astropy.units as u
import numpy as np
Read in input images¶
We first read in the counts cube and sum over the energy axis:
[2]:
counts = Map.read(
"$GAMMAPY_DATA/fermi-3fhl-gc/fermi-3fhl-gc-counts-cube.fits.gz"
)
background = Map.read(
"$GAMMAPY_DATA/fermi-3fhl-gc/fermi-3fhl-gc-background-cube.fits.gz"
)
exposure = Map.read(
"$GAMMAPY_DATA/fermi-3fhl-gc/fermi-3fhl-gc-exposure-cube.fits.gz"
)
# unit is not properly stored on the file. We add it manually
exposure.unit = "cm2s"
psf = EnergyDependentTablePSF.read(
"$GAMMAPY_DATA/fermi-3fhl-gc/fermi-3fhl-gc-psf-cube.fits.gz"
)
psfmap = PSFMap.from_energy_dependent_table_psf(psf)
edisp = EDispKernelMap.from_diagonal_response(
energy_axis=counts.geom.axes["energy"],
energy_axis_true=exposure.geom.axes["energy_true"],
)
dataset = MapDataset(
counts=counts,
background=background,
exposure=exposure,
psf=psfmap,
name="fermi-3fhl-gc",
edisp=edisp,
)
WARNING: FITSFixedWarning: 'datfix' made the change 'Set DATE-REF to '1858-11-17' from MJD-REF.
Set MJD-OBS to 54682.655278 from DATE-OBS.
Set MJD-END to 57236.967546 from DATE-END'. [astropy.wcs.wcs]
WARNING: FITSFixedWarning: 'datfix' made the change 'Set DATE-REF to '2001-01-01T00:01:04.184' from MJD-REF.
Set MJD-OBS to 54682.655283 from DATE-OBS.
Set MJD-END to 57236.967546 from DATE-END'. [astropy.wcs.wcs]
Adaptive smoothing¶
For visualisation purpose it can be nice to look at a smoothed counts image. This can be performed using the adaptive smoothing algorithm from Ebeling et al. (2006).
In the following example the threshold
argument gives the minimum significance expected, values below are clipped.
[3]:
%%time
scales = u.Quantity(np.arange(0.05, 1, 0.05), unit="deg")
smooth = ASmoothMapEstimator(
threshold=3, scales=scales, energy_edges=[10, 500] * u.GeV
)
images = smooth.run(dataset)
CPU times: user 981 ms, sys: 410 ms, total: 1.39 s
Wall time: 1.39 s
[4]:
plt.figure(figsize=(15, 5))
images["flux"].plot(add_cbar=True, stretch="asinh");
TS map estimation¶
The Test Statistic, TS = 2 ∆ log L (Mattox et al. 1996), compares the likelihood function L optimized with and without a given source. The TS map is computed by fitting by a single amplitude parameter on each pixel as described in Appendix A of Stewart (2009). The fit is simplified by finding roots of the derivative of the fit statistics (default settings use Brent’s method).
We first need to define the model that will be used to test for the existence of a source. Here, we use a point source.
[5]:
spatial_model = PointSpatialModel()
spectral_model = PowerLawSpectralModel(index=2)
model = SkyModel(spatial_model=spatial_model, spectral_model=spectral_model)
[6]:
%%time
estimator = TSMapEstimator(
model,
kernel_width="1 deg",
selection_optional=[],
energy_edges=[10, 500] * u.GeV,
)
maps = estimator.run(dataset)
CPU times: user 11.5 s, sys: 115 ms, total: 11.6 s
Wall time: 11.5 s
Plot resulting images¶
[7]:
plt.figure(figsize=(15, 5))
maps["sqrt_ts"].plot(add_cbar=True);
[8]:
plt.figure(figsize=(15, 5))
maps["flux"].plot(add_cbar=True, stretch="sqrt", vmin=0);
/Users/adonath/software/anaconda3/envs/gammapy-dev/lib/python3.7/site-packages/matplotlib/colors.py:581: RuntimeWarning: invalid value encountered in less
xa[xa < 0] = -1
/Users/adonath/software/anaconda3/envs/gammapy-dev/lib/python3.7/site-packages/matplotlib/colors.py:581: RuntimeWarning: invalid value encountered in less
xa[xa < 0] = -1
[9]:
plt.figure(figsize=(15, 5))
maps["niter"].plot(add_cbar=True);
Source candidates¶
Let’s run a peak finder on the sqrt_ts
image to get a list of point-sources candidates (positions and peak sqrt_ts
values). The find_peaks
function performs a local maximun search in a sliding window, the argument min_distance
is the minimum pixel distance between peaks (smallest possible value and default is 1 pixel).
[10]:
sqrt_ts_image = maps["sqrt_ts"].get_image_by_idx((0,))
sources = find_peaks(sqrt_ts_image, threshold=5, min_distance="0.25 deg")
nsou = len(sources)
sources
[10]:
value | x | y | ra | dec |
---|---|---|---|---|
deg | deg | |||
float64 | int64 | int64 | float64 | float64 |
32.242 | 200 | 99 | 266.41449 | -28.97054 |
27.846 | 52 | 60 | 272.43197 | -23.54282 |
15.213 | 32 | 98 | 271.16056 | -21.74479 |
14.176 | 69 | 93 | 270.40919 | -23.47797 |
13.923 | 80 | 92 | 270.15899 | -23.98049 |
9.7645 | 273 | 119 | 263.18257 | -31.52587 |
8.7988 | 124 | 102 | 268.46711 | -25.63326 |
7.3528 | 123 | 134 | 266.97596 | -24.77174 |
6.8117 | 193 | 19 | 270.59696 | -30.69138 |
6.2469 | 152 | 148 | 265.48068 | -25.64323 |
5.8947 | 230 | 86 | 266.15140 | -30.58926 |
5.6621 | 127 | 12 | 272.77351 | -27.97934 |
5.6566 | 251 | 139 | 262.90685 | -30.05853 |
5.478 | 181 | 95 | 267.17020 | -28.26173 |
5.4309 | 214 | 83 | 266.78188 | -29.98429 |
5.1818 | 57 | 49 | 272.82739 | -24.02653 |
5.0702 | 156 | 132 | 266.12148 | -26.23306 |
5.0545 | 93 | 80 | 270.37773 | -24.84233 |
[11]:
# Plot sources on top of significance sky image
plt.figure(figsize=(15, 5))
_, ax, _ = maps["sqrt_ts"].plot(add_cbar=True)
ax.scatter(
sources["ra"],
sources["dec"],
transform=plt.gca().get_transform("icrs"),
color="none",
edgecolor="w",
marker="o",
s=600,
lw=1.5,
);
Note that we used the instrument point-spread-function (PSF) as kernel, so the hypothesis we test is the presence of a point source. In order to test for extended sources we would have to use as kernel an extended template convolved by the PSF. Alternatively, we can compute the significance of an extended excess using the Li & Ma formalism, which is faster as no fitting is involve.
What next?¶
In this notebook, we have seen how to work with images and compute TS and significance images from counts data, if a background estimate is already available.
Here’s some suggestions what to do next:
Look how background estimation is performed for IACTs with and without the high-level interface in analysis_1 and analysis_2 notebooks, respectively
Learn about 2D model fitting in the modeling 2D notebook
find more about Fermi-LAT data analysis in the fermi_lat notebook
Use source candidates to build a model and perform a 3D fitting (see analysis_3d, analysis_mwl notebooks for some hints)
[ ]: