Python tv_default_loaderの例

コード例 #1

ファイル: parallelfolder.py プロジェクト: yzcck/dissect

def default_loader(filename):
    '''
    Handles both numpy files and image formats.
    '''
    if filename.endswith('.npy') or filename.endswith('.NPY'):
        return numpy.load(filename).view(ndarray)
    elif filename.endswith('.npz') or filename.endswith('.NPZ'):
        return numpy.load(filename)
    else:
        return tv_default_loader(filename)

コード例 #2

def default_loader(filename):
    '''
    Handles both numpy files and image formats.
    '''
    if filename.endswith('.npy'):
        return numpy.load(filename).view(ndarray)
    elif filename.endswith('.npz'):
        return numpy.load(filename)
    else:
        try:
            return tv_default_loader(filename)
        except:
            print('Error in', filename)
            # Some images the high-speed loader can't load but PIL can.
            im = Image.open(filename)
            im.load()
            return im