我正在做基本的烤宽面条示例: 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参数赋予任何大小,则代码会在尝试再次编译神经网络时卡住。
答案 0 :(得分:0)
好的,我想我可以在这里看到一些问题,不知道你遇到了哪一个......
在read_images1()
中,X是numpy数组的python列表。它在哪里被转换为单个numpy数组?尝试添加X = numpy.asarray(X)
。您还需要将其重新整形为(n_images,n_channels,width,height),其中我假设n_channels = 1用于灰度。网络预计4D输入不是3D。
此代码list_of_labels = list(xrange(max(label) + 1)); classes = len(list_of_labels)
假定标签是从0到N的连续数字。是吗?
您的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, ...)
此外,如果它在build_mlp()
期间挂起,那么显然问题不在于您如何阅读图像。尝试将build_mlp()
的原始版本与您的图片一起使用。尝试自己运行它。跳过图像读取,只需使用常量参数调用build_mlp()
。