Commit 36db2fcc authored by Pierre Paleo's avatar Pierre Paleo
Browse files

Integrate in fullfield pipeline

parent 79b5134e
Pipeline #26949 passed with stages
in 2 minutes and 39 seconds
......@@ -4,8 +4,8 @@ from .logger import LoggerOrPrint
from ..utils import check_supported
from ..io.reader import ChunkReader
from ..preproc.ccd import FlatField, Log
from ..preproc.phase import PaganinPhaseRetrieval
from ..preproc.double_flatfield import DoubleFlatField
from ..preproc.phase import PaganinPhaseRetrieval
from ..preproc.sinogram import SinoProcessing
from ..misc.unsharp import UnsharpMask
from ..resources.utils import is_hdf5_extension
......@@ -20,11 +20,11 @@ class FullFieldPipeline:
"""
FlatFieldClass = FlatField
DoubleFlatFieldClass = DoubleFlatField
PaganinPhaseRetrievalClass = PaganinPhaseRetrieval
UnsharpMaskClass = UnsharpMask
SinoProcessingClass = SinoProcessing
MLogClass = Log
DoubleFlatFieldClass = DoubleFlatField
FBPClass = None # For now we don't have a plain python/numpy backend for reconstruction
def __init__(self, process_config, sub_region, logger=None, extra_options=None):
......
from math import ceil
import numpy as np
from ..preproc.ccd_cuda import CudaFlatField, CudaLog
from ..preproc.double_flatfield_cuda import CudaDoubleFlatField
from ..preproc.phase_cuda import CudaPaganinPhaseRetrieval
from ..preproc.sinogram_cuda import CudaSinoProcessing
from ..preproc.sinogram import SinoProcessing
......@@ -19,6 +20,7 @@ class CudaFullFieldPipeline(FullFieldPipeline):
"""
FlatFieldClass = CudaFlatField
DoubleFlatFieldClass = CudaDoubleFlatField
PaganinPhaseRetrievalClass = CudaPaganinPhaseRetrieval
UnsharpMask = CudaUnsharpMask
SinoProcessingClass = CudaSinoProcessing
......
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