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)
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