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