@@ -60,4 +60,4 @@ If you would like use the native training model of deepCR, which are obtained fr
# Bugs
Please contact **_HU Yi_** (huyi.naoc@gmail.com) to report bugs.
Bugs can be reported and filed at https://csst-tb.bao.ac.cn/code/huyi/csst-commom-crmask/-/issues. Alternatively, you can contact **_HU Yi_** (huyi.naoc@gmail.com) to report bugs.
Instantiation of CRMask with specified model configuration.
Parameters
----------
__________
obj : string, Path, astropy.io.fits.HDUList, numpy.ndarray, astropy.nddata.CCDData or list of string
if model is ``deepCR``, ``lacosmic``, obj is input image to be cosmic-ray masked
if model is ``deepCR_train``, obj is input training and validating images
if model is ``deepCR``, ``lacosmic``, obj is input image to be cosmic-ray masked, if model is ``deepCR_train``, obj is input training and validating images
flag : (optional) string, Path, astropy.io.fits.HDUList, numpy.ndarray or astropy.nddata.CCDData
flag image(s), default is None
mask : (optional) string or list of string
...
...
@@ -68,6 +67,7 @@ class CRMask:
config_path : (optional) string
configuration file path, default is ``./crmask.ini``
"""
self.model=model
ifmodel=='deepCR_train':
self.image_sets=obj
...
...
@@ -450,8 +450,9 @@ class CRMask:
Only returned if `fill_flag` is True
cosmic ray cleaned image.
Examples
-------
--------
>>> from crmask import CRMask
>>> crobj = CRMask('xxxx.fits', 'deepCR')
>>> crobj.cr_mask()
...
...
@@ -710,21 +711,44 @@ class CRMask:
defcr_train(self):
"""
Training models, only support ``deepCR_train``. It will generate pytorch's *.pth file.
The train is very painful and time consuming, do NOT use it in pipelines.
Training models, only support ``deepCR_train``. It will generate pytorch's \*.pth file. The train is very painful and time consuming, do NOT use it in pipelines.