Commit c9ab289f authored by Hu Yi's avatar Hu Yi
Browse files

Update crmask.py

parent 0fce31a8
......@@ -31,7 +31,7 @@ import sys
__all__ = ['CRMask']
class CRMask:
def __init__(self, obj, flag = None, mask = None, clean = None, sky = None, save_flag = True, update_flag = True, save_name = None, flag_suffix = 'flg', clean_suffix = 'crclean', append_flag = False, mask_suffix = 'crmask', model = 'deepCR', fill_flag = True, fill_method = 'inpainting', gpu_flag = False, config_path = 'MSC_crmask.ini', **kwargs):
def __init__(self, obj, flag = None, mask = None, clean = None, sky = None, save_flag = True, update_flag = True, save_name = None, flag_suffix = 'flg', clean_suffix = 'crclean', append_flag = False, mask_suffix = 'crmask', model = 'deepCR', fill_flag = True, fill_method = 'inpainting', gpu_flag = False, config_path = 'crmask.ini', **kwargs):
"""
Instantiation of CRMask with specified model configuration.
Parameters
......@@ -66,7 +66,7 @@ class CRMask:
gpu_flag : (optional) boolean
whether use GPU, default is False
config_path : (optional) string
configuration file path, default is ``../conf/MSC_crmask.ini``
configuration file path, default is ``./crmask.ini``
"""
self.model = model
if model == 'deepCR_train':
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment