Commit 9978aeeb authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

added photometry routine from Hu Zou

parent 1f8e6306
Loading
Loading
Loading
Loading

csst/msc/phot.py

0 → 100644
+31 −0
Original line number Diff line number Diff line
from pathlib import Path

import numpy as np
from ccdproc import cosmicray_lacosmic
from deepCR import deepCR

from ..core.processor import CsstProcessor, CsstProcStatus
from ._photometry.csst_photometry import do_phot
from ..msc import CsstMscImgData
from .. import PACKAGE_PATH


# DEEPCR_MODEL_PATH = PACKAGE_PATH + "/msc/deepcr_model/CSST_2021-12-30_CCD23_epoch20.pth"


class CsstMscInPhotometryProc(CsstProcessor):

    def __init__(self):
        super(CsstProcessor, self).__init__()

    def prepare(self, **kwargs):
        pass

    def cleanup(self):
        pass

    def run(self, fl_in, out_dir):
        # assert len(fl_in) == len(fl_out)
        for i in range(len(fl_in)):
            do_phot(fl_in[i], outdir=out_dir)
        return