Was*_*tay 4 python regression conv-neural-network tensorflow
我正在研究一个问题,根据奶牛的图像预测奶牛有多胖的分数。我应用了 CNN 来估计 0-5 之间的值(我拥有的数据集仅包含 2.25 和 4 之间的值)我使用了 4 个 CNN 层和 3 个隐藏层。
我实际上有两个问题:1/我得到了 0.05 的训练误差,但是在 3-5 个时期之后,验证误差仍然保持在 0.33 左右。2/ 我的神经网络预测的值在 2.9 到 3.3 之间,与数据集范围相比太窄了。正常吗?
我怎样才能改进我的模型?
model = tf.keras.models.Sequential([
    tf.keras.layers.Conv2D(16, (3,3), activation='relu', input_shape=(512, 424,1)),
    tf.keras.layers.MaxPooling2D(2, 2),
    tf.keras.layers.Conv2D(32, (3,3), activation='relu'),
    tf.keras.layers.MaxPooling2D(2, 2),
    tf.keras.layers.Conv2D(32, (3,3), activation='relu'),
    tf.keras.layers.MaxPooling2D(2, 2),
    tf.keras.layers.Conv2D(64, (3,3), activation='relu'),
    tf.keras.layers.MaxPooling2D(2,2),
    tf.keras.layers.Flatten(input_shape=(512, 424)),
    tf.keras.layers.Dense(256, activation=tf.nn.relu),
    tf.keras.layers.Dense(128, activation=tf.nn.relu),
    tf.keras.layers.Dense(64, activation=tf.nn.relu),
    tf.keras.layers.Dense(1, activation='linear')
])
Run Code Online (Sandbox Code Playgroud)
学习曲线:

预言:

这似乎是过度拟合的情况。你可以
Shuffle的Data,用shuffle=True在cnn_model.fit。代码如下所示:
history = cnn_model.fit(x = X_train_reshaped,
                        y = y_train,
                        batch_size = 512,
                        epochs = epochs, callbacks=[callback],
                        verbose = 1, validation_data = (X_test_reshaped, y_test),
                        validation_steps = 10, steps_per_epoch=steps_per_epoch, shuffle = True)
使用Early Stopping. 代码如下所示
callback = tf.keras.callbacks.EarlyStopping(monitor='val_loss', patience=15)
使用正则化。正则化代码如下所示(您也可以尝试 l1 正则化或 l1_l2 正则化):
from tensorflow.keras.regularizers import l2
Regularizer = l2(0.001)
cnn_model.add(Conv2D(64,3, 3, input_shape = (28,28,1), activation='relu', data_format='channels_last', 
                    activity_regularizer=Regularizer, kernel_regularizer=Regularizer))
cnn_model.add(Dense(units = 10, activation = 'sigmoid', 
                    activity_regularizer=Regularizer, kernel_regularizer=Regularizer))
您可以尝试使用BatchNormalization.
使用ImageDataGenerator. 有关更多信息,请参阅此链接。
如果像素不是Normalized,将像素值除以255也有帮助。
最后,如果仍然没有变化,您可以尝试使用Pre-Trained ModelslikeResNet或VGG Net等。
|   归档时间:  |  
           
  |  
        
|   查看次数:  |  
           320 次  |  
        
|   最近记录:  |