__init__.py 文件源码

def _mnist_raw_labels():
    fname = resource_filename('sudokuextract.data', "train-labels-idx1-ubyte.gz")
    if resource_exists('sudokuextract.data', "train-labels-idx1-ubyte.gz"):
        f = gzip.open(fname, mode='rb')
        data = f.read()
        f.close()
    else:
        sio = StringIO(urlopen(_url_to_mnist_train_labels).read())
        sio.seek(0)
        f = gzip.GzipFile(fileobj=sio, mode='rb')
        data = f.read()
        f.close()
        try:
            sio.seek(0)
            with open(fname, 'wb') as f:
                f.write(sio.read())
        except:
            pass

    correct_magic_number = 2049
    magic_number = _toS32(data[:4])
    if magic_number != correct_magic_number:
        raise ValueError("Error parsing labels file. Read magic number {0} != {1}!".format(
            magic_number, correct_magic_number))
    n_labels = _toS32(data[4:8])
    return np.fromstring(data[8:], 'uint8')