zhangzzk / forklens

DL weak lensing shear

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Forklens

A deep-learning-based method to measure weak gravitational lensing signal.


d1a4058a-979b-4963-8aa3-11e04e9c8537

(Image by OpenAI)


This project is originally motivated to measure galaxy shapes (shear) and in the meantime correct the smearing of the point spread function (PSF, an effect from either/both the atmosphere and optical instrument). The code contains a custom CNN achitecture which has two input branches, fed with the observed galaxy image and PSF image, predicting several features of the galaxy (shapes, magnitude, size, etc.). Simulation in the code is built directly upon Galsim.

To install

cd forklens

python setup.py install

Check some example tests in ./examples.

Requirements

Python==3.9.7, Pytorch==1.11.0+cu113, Galsim==2.3.4

About

DL weak lensing shear


Languages

Language:Python 100.0%