基于python实现的BM3D算法

这是一本基于python实现的BM3D去噪算法，值得学习一下

“““PyBM3D packaging and distribution.“““
import os
from setuptools import setup Extension
from setuptools.command.build_ext import build_ext as _build_ext


class CustomBuildExt（_build_ext）:
    “““Custom build extension class.“““

    def __init__（self *args **kwargs）:
        “““Extends default class constructor.“““
        # _build_ext is not a new-style （object） class which is required for
        # super to work
        _build_ext.__init__（self *args **kwargs）
        self.cuda_config = self.load_cuda_config（）

    def build_extensions（self）:
        “““Extends default build_extensions method.

        Further customizes the compiler to add C file specific compile
        arguments and support nvcc compilation of *.cu CUDA files.“““

        self.customize_compiler_for_c_args_and_nvcc（）
        _build_ext.build_extensions（self）

    def finalize_options（self）:
        “““Extends default finalize_options method.

        Injects NumPy‘s C include directories into the Cython compilation. This
        is done after NumPy was installed through the setuptools setup_requires
        argument which removes NumPy from the necessary preinstalled
        packages.“““

        _build_ext.finalize_options（self）
        # prevent numpy from thinking it is still in its setup process
        __builtins__.__NUMPY_SETUP__ = False
        import numpy
        self.include_dirs.append（numpy.get_include（））

    def customize_compiler_for_c_args_and_nvcc（self）:
        “““Customize the compiler.

        The customization adds C file specific compile
        arguments and support for nvcc compilation of *.cu CUDA files.“““

        self.compiler.src_extensions.append（‘.cu‘）

        # save references to the default compiler_so and _comple methods
        default_compiler_so = self.compiler.compiler_so
        super = self.compiler._compile

        # now redefine the _compile method. This gets executed for each
        # object but distutils doesn‘t have the ability to change compilers
        # based on source extension: we add it.
        def _compile（obj src ext cc_args extra_postargs pp_opts）:
            if os.path.splitext（src）[1] == ‘.cu‘:
                # use the nvcc for *.cu files
                self.compiler.set_executable（‘compiler_so‘
                                             self.cuda_config[‘nvcc‘]）
                postargs = extra_postargs[‘nvcc‘]
            else:
                postargs = extra_postargs[‘unix‘]

                # add C file specific compile arguments
                if os.path.splitext（src）[1] == ‘.c‘:
                    postargs = postargs + extra_postargs[‘c_args‘]

            super（obj src ext cc_args postargs pp_opts）
            # reset the default compiler_so
            self.compiler.compiler_so = default_compiler_so

        # inject our redefined _compile method into the class
        self.compiler._compile = _compile

    @staticmethod
    def load_cuda_config（）:
        “““Locate the CUDA env

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2017-10-09 10:23  pybm3d-0.2.1\
     文件         278  2017-10-09 10:23  pybm3d-0.2.1\.gitignore
     文件          84  2017-10-09 10:23  pybm3d-0.2.1\.gitmodules
     文件         139  2017-10-09 10:23  pybm3d-0.2.1\.scrutinizer.yml
     文件         841  2017-10-09 10:23  pybm3d-0.2.1\.travis.yml
     文件       35142  2017-10-09 10:23  pybm3d-0.2.1\LICENSE
     文件          23  2017-10-09 10:23  pybm3d-0.2.1\MANIFEST.in
     文件        3939  2017-10-09 10:23  pybm3d-0.2.1\README.rst
     目录           0  2017-10-09 10:23  pybm3d-0.2.1\bm3d_src\
     目录           0  2017-10-09 10:23  pybm3d-0.2.1\pybm3d\
     文件          90  2017-10-09 10:23  pybm3d-0.2.1\pybm3d\__init__.py
     文件        4594  2017-10-09 10:23  pybm3d-0.2.1\pybm3d\bm3d.pyx
     文件         164  2017-10-09 10:23  pybm3d-0.2.1\setup.cfg
     文件        6004  2017-10-09 10:23  pybm3d-0.2.1\setup.py
     目录           0  2017-10-09 10:23  pybm3d-0.2.1\tests\
     文件        2639  2017-10-09 10:23  pybm3d-0.2.1\tests\test_bm3d.py
     文件         502  2017-10-09 10:23  pybm3d-0.2.1\tox.ini