尝试在Lasagne基本示例(略微修改)上使用自己的数据集来制作用于人脸识别的神经网络

时间:2016-04-11 18:42:10

标签: python machine-learning neural-network theano lasagne

我正在做基本的烤宽面条示例: https://github.com/Lasagne/Lasagne/blob/master/examples/mnist.py

我将其与另一个类似的例子进行了轻微修改。

我试图运行CNN模型,我已经为CNN def添加了一些额外的输入,但它不应该有任何不同。同样将示例中输入图层的默认值28更改为60(高度和宽度),稍后在代码中使用类,但代码"挂起"在最后一个网络线上,意味着代码仍在运行,但没有任何事情发生。 Output when running the code. input_var在mainloop中定义如下:

input_var = T.tensor4('input_var')

其余代码:

def build_cnn(classes, height, width, input_var=None):

    print("Input layer, with height: {}, width: {} and input var: {}".format(height, width, input_var))

    network = lasagne.layers.InputLayer(shape = (None, 1, height, width),
                                    input_var=input_var)


    print("Convolutional layer with 32 kernels of size 5x5")
    network = lasagne.layers.Conv2DLayer(network,
                                         num_filters = 32,
                                         filter_size = (5, 5),
                                         nonlinearity = lasagne.nonlinearities.rectify,
                                         W = lasagne.init.HeNormal(gain = 'relu')) 

修改 好的,根据我迄今为止所尝试的内容,似乎是我自己的数据集就是问题所在。我重塑了我的数据集以匹配MNIST数据集,其中ex。 X_train的形状为[图像,通道,高度,宽度]。其中channel = 1和height,width = 60.检索这些的代码是:

def load_images():
    dataset_path = os.path.abspath("C:/Users/laende/Dropbox/Skole UiS/4. semester/Master/Master/data/test_database")
    [bilder, label, names] = read_images1(dataset_path, (28, 28))
    label = np.array(label)

    bilder = bilder / np.float32(256)
    bilder = bilder[:, newaxis, :, :]

    X_train1, X_test1, Y_train1, Y_test1 = train_test_split(bilder, label, test_size = 0.2)

    list_of_labels = list(xrange(max(label) + 1))
    classes = len(list_of_labels)

    return X_train1, X_test1, Y_train1, Y_test1, classes

其中read_images1是:

 def read_images1(path, sz = None, channel = None):
    c = 0
    X = []
    y = []
    folder_names = []
    for dirname, dirnames, filenames, in os.walk(path):
        for subdirname in dirnames:
            subject_path = os.path.join(dirname, subdirname)
            folder_names.append(subdirname)
            for filename in os.listdir(subject_path):
                try:
                    im = cv2.imread(os.path.join(subject_path, filename),     cv2.IMREAD_GRAYSCALE)

                    if (sz is not None):
                        im = cv2.resize(im, sz)

                    X.append(np.asarray(im, dtype = np.uint8))
                    y.append(c)

                except IOError, (errno, strerror):
                    print "I/O error ({0]): {1}".format(errno, strerror)
                except:
                    print "unexpected error:", sys.exc_info()[0]
                    raise
            c = c + 1
    return [X, y, folder_names]

运行的主要代码:

def main(model='mlp', num_epochs=100):
    # Load the dataset

    print("Loading data...")
    mnist = 1
    if mnist == 1:
        classes = 10
        X_train, y_train, X_val, y_val, X_test, y_test = load_dataset()

        dataset = {
            'train': {'X': X_train, 'y': y_train},
            'test': {'X': X_test, 'y': y_test}}
        shape = dataset['train']['X'][0].shape

    else:
        X_train, X_test, y_train, y_test, classes = load_images()

        dataset = {
            'train': {'X': X_train, 'y': y_train},
            'test': {'X': X_test, 'y': y_test}}
        shape = dataset['train']['X'][0].shape

    input_var = T.tensor4('inputs')
    target_var = T.ivector('targets')

    print("Building model and compiling functions...")
    if model == 'mlp':
        network = build_mlp(height=int(shape[1]),
                            width=int(shape[2]),
                            channel=int(shape[0]),
                            classes=int(classes),
                            input_var=input_var)

如果mnist = 1(在main中)代码运行正常,如果我尝试使用我自己的数据集,它会停留在build_mlp中(类似于cnn的原始问题):

def build_mlp(classes, channel, height, width, input_var=None):

    neurons = int(height * width)

    network = lasagne.layers.InputLayer(shape = (None, channel, height, width),
                                 input_var=input_var)

    network = lasagne.layers.DropoutLayer(network, p = 0.2)

   #Code gets stuck on this point, running forever, doing nothing.
   #No error messages received either.
    network = lasagne.layers.DenseLayer(
        network,
        num_units = neurons,
        nonlinearity = lasagne.nonlinearities.rectify,
        W = lasagne.init.GlorotUniform())

编辑2: 经过一段时间的努力,我发现在read_images1()中完成的图像大小调整导致了问题:

 def read_images1(path, sz = None, channel = None):
    c = 0
    X = []
    y = []
    folder_names = []
    for dirname, dirnames, filenames, in os.walk(path):
        for subdirname in dirnames:
            subject_path = os.path.join(dirname, subdirname)
            folder_names.append(subdirname)
            for filename in os.listdir(subject_path):
                try:
                    im = cv2.imread(os.path.join(subject_path, filename),     cv2.IMREAD_GRAYSCALE)
                    #This part caused the problems. 
                    if (sz is not None):
                        im = cv2.resize(im, sz)

                    X.append(np.asarray(im, dtype = np.uint8))
                    y.append(c)

                except IOError, (errno, strerror):
                    print "I/O error ({0]): {1}".format(errno, strerror)
                except:
                    print "unexpected error:", sys.exc_info()[0]
                    raise
            c = c + 1
    return [X, y, folder_names]

如果我没有通过任何调整大小并使用文件夹中的默认图像大小,神经网络就能够编译。有谁知道为什么?我将read_images1()更新为:

 def read_images1(path, sz = None, na = False):
    """

    :param path: sti til mappe med underliggende mapper tilhørende personer.
    :param sz: Størrelse på bildefilene
    :return: returnerer liste av bilder, labels og navn
    """
    c = 0
    X = []
    y = []
    folder_names = []
    for dirname, dirnames, filenames, in os.walk(path):
        for subdirname in dirnames:
            subject_path = os.path.join(dirname, subdirname)
            folder_names.append(subdirname)
            for filename in os.listdir(subject_path):
                try:
                    im = cv2.imread(os.path.join(subject_path, filename), cv2.IMREAD_GRAYSCALE)

                    if (sz is not None):
                        im = cv2.resize(im, dsize=sz, interpolation = cv2.INTER_LANCZOS4)

                    if (na == True):
                        im = im[newaxis, :, :]


                    X.append(np.asarray(im, dtype = np.uint8))
                    y.append(c)

                except IOError, (errno, strerror):
                    print "I/O error ({0]): {1}".format(errno, strerror)
                except:
                    print "unexpected error:", sys.exc_info()[0]
                    raise
            c = c + 1
    return [X, y, folder_names] 

如果我用sz = None和na = True运行程序,那么它可以工作。如果给sz参数赋予任何大小,则代码会在尝试再次编译神经网络时卡住。

1 个答案:

答案 0 :(得分:0)

好的,我想我可以在这里看到一些问题,不知道你遇到了哪一个......

  1. read_images1()中,X是numpy数组的python列表。它在哪里被转换为单个numpy数组?尝试添加X = numpy.asarray(X)。您还需要将其重新整形为(n_images,n_channels,width,height),其中我假设n_channels = 1用于灰度。网络预计4D输入不是3D。

  2. 此代码list_of_labels = list(xrange(max(label) + 1)); classes = len(list_of_labels)假定标签是从0到N的连续数字。是吗?

  3. 您的build_mlp(classes, height, width, input_var=None)与原始示例build_mlp(input_var=None)有很大不同。最初的例子清楚地起作用,所以无论错误与差异有什么关系。一个很大的区别是你继续分配到同一个变量,例如network = lasagne.layers.DenseLayer(network, ...),其中原始变量对每个层都有不同的变量l_hid1 = lasagne.layers.DenseLayer(l_in_drop, ...)

  4. 此外,如果它在build_mlp()期间挂起,那么显然问题不在于您如何阅读图像。尝试将build_mlp()的原始版本与您的图片一起使用。尝试自己运行它。跳过图像读取,只需使用常量参数调用build_mlp()