"vscode:/vscode.git/clone" did not exist on "f6f759fc5fb4868125b8a25c28ce96d2c0980ef7"
__init__.py 836 Bytes
Newer Older
1
2
3
4
from torchvision import models
from torchvision import datasets
from torchvision import transforms
from torchvision import utils
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32


_image_backend = 'PIL'


def set_image_backend(backend):
    """
    Specifies the package used to load images.

    Options are 'PIL' and 'accimage'. The :mod:`accimage` package uses the
    Intel IPP library. It is generally faster than PIL, but does not support as
    many operations.

    Args:
        backend (string): name of the image backend
    """
    global _image_backend
    if backend not in ['PIL', 'accimage']:
        raise ValueError("Invalid backend '{}'. Options are 'PIL' and 'accimage'"
                         .format(backend))
    _image_backend = backend


def get_image_backend():
    """
    Gets the name of the package used to load images
    """
    return _image_backend