61a0663a45beb2e59f6a549c51d55940c80734e3,niftynet/io/image_as_nibabel.py,,,#,11

Before Change

from PIL import Image

try:
    // Check whether scikit-image is installed in the system and use it.
    // It loads images slightly faster then PIL.
    // Only use scikit-image if installed version is 0.13.0 or newer
    from niftynet.utilities.util_import import require_module
    SKIO = require_module("skimage.io", (0, 13, 0))
    USE_SKIMAGE = True
    tf.logging.info("+++ Using SKIMAGE as Image Loading backend")
except (ImportError, AssertionError):
    tf.logging.info("+++ Using PIL as Image Loading backend")
    USE_SKIMAGE = False


def image2nibabel(filename):
    
    Loads a RGB or Grayscale Image from a file and stores it in a 5D array,
    moving the color channels to the last axis for color images.

After Change

EXTERNAL_LOADERS = [
    dict(name="opencv", module="cv2", method="imread", kwargs=dict(flags=-1)),
    dict(name="skimage", module="skimage.io", method="imread", version="0.13"),
    dict(name="pillow", module="PIL.Image", method="open")
]

AVAILABLE_LOADER = OrderedDict()

for loader in EXTERNAL_LOADERS:
    try:
        // Default params
        min_ver = loader.get("version", None)
        args = loader.get("args", tuple())
        kwargs = loader.get("kwargs", dict())
        // Retrieve external function
        external_module = require_module(loader["module"], min_version=min_ver)
        external_function = getattr(external_module, loader["method"])
        // Save loader params
        loader_dict = dict(fn=external_function, args=args, kwargs=kwargs)
    except (ImportError, AssertionError, AttributeError):
        continue

    AVAILABLE_LOADER[loader["name"]] = loader_dict


tf.logging.info("+++ Available Image Loaders {}:"
                 .format(list(AVAILABLE_LOADER.keys())))

Instances