NAN的损失,精度为0,我尝试了一切! (CNN)

时间:2020-05-09 00:51:03

标签: python tensorflow deep-learning conv-neural-network nan

我正在尝试创建一个可以识别字母(26个类)的NN。对于冗长的帖子,我深表歉意,但是我已经包括了所有相关代码,以使其尽可能清晰。最后,我解释了这个问题。

以下块是我正确命名路径,标准化/标准化图像并准备进行训练的地方。

X = [] # Image data
y = [] # Labels

datagen = ImageDataGenerator(samplewise_center=True)

for path in imagepaths:
  img = cv2.imread(path)
  img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) 
  img = cv2.resize(img, (200, 200))
  img = image.img_to_array(img)
  img = datagen.standardize(img)
  X.append(img)

  # Processing label in image path
  category = path.split("\\")[1]
  #print(category)
  split = (category.split("_"))     
  if int(split[0]) == 0:
    label = int(split[1])
  else:
    label = int(split[0])
  y.append(label)

# Turn X and y into np.array to speed up train_test_split
X = np.array(X, dtype="uint8")
X = X.reshape(len(imagepaths), 200, 200, 1) 
y = np.array(y)

创建测试集。

ts = 0.3 
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=ts, random_state=42)

创建模型。密度为26,每个字母一个输出,大小为200,200,1以匹配输入:

model = Sequential()
model.add(Conv2D(32, (3, 3), activation='relu', input_shape=(200, 200, 1))) 
model.add(MaxPooling2D((2, 2)))
model.add(BatchNormalization())
model.add(Dropout(0.5))
model.add(Conv2D(64, (3, 3), activation='relu')) 
model.add(MaxPooling2D((2, 2)))
model.add(BatchNormalization())
model.add(Dropout(0.5))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(MaxPooling2D((2, 2)))
model.add(BatchNormalization())
model.add(Dropout(0.5))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dense(26, activation='softmax'))

模型编译器适合:

model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy']) 
model.fit(X_train, y_train, epochs=1, batch_size=64, verbose=1, validation_data=(X_test, y_test))

问题出在这里,我的model.fit的输出是:

Train on 54600 samples, validate on 23400 samples
Epoch 1/1
54600/54600 [==============================] - 79s 1ms/step - loss: nan - accuracy: 1.8315e-05 - val_loss: nan - val_accuracy: 0.0000e+00

我知道这可能不是很高的准确性,也可能一无所获,但是为什么损失会减少呢?我在其他地方发布了信息,并首先被告知要对我的数据进行规范化(我对此信息进行了修复)。然后我被告知,我的数据集可能已损坏或泄漏-情况并非如此,因为当我一次不写26个字母时,它就可以正常工作。 (意思是我使用字母A-J,密集= 10等对代码进行了测试),并获得了约95%的高精度。

感谢您的帮助,我已经为此花了好几个小时了!

0 个答案:

没有答案